Open CaoYiwei opened 1 year ago
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])
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])