DeepMatch icon indicating copy to clipboard operation
DeepMatch copied to clipboard

utils.py的inner product实现有问题

Open CaoYiwei opened this issue 3 years ago • 1 comments

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])

CaoYiwei avatar Nov 08 '22 06:11 CaoYiwei

是的,没有加axis,这样算的是一个value

sosixyz avatar Feb 23 '25 03:02 sosixyz