New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

utils.py的inner product实现有问题 #93

Open

CaoYiwei opened this issue Nov 8, 2022 · 0 comments

CaoYiwei commented Nov 8, 2022

inner product是通过tf.reduce_sum实现的，reduce时候没有指定维度，建议默认axis=-1

当前0.3.1版本dssm模型无法构建，按如下方式修改后可以正常训练。

def inner_product(x, y, temperature=1.0, axis=-1):
return Lambda(lambda x: tf.reduce_sum(tf.multiply(x[0], x[1]), axis) / temperature)([x, y])

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment