Skip to content

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

Open
@CaoYiwei

Description

@CaoYiwei

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions