运行报错怎么回事?
LGY-111 opened this issue · comments
改为:
similarities = similarities - tf.eye(K.shape(y_pred)[0]) * 1e12 # 排除对角线
a bert for retrieval and generation
LGY-111 opened this issue · comments
改为:
similarities = similarities - tf.eye(K.shape(y_pred)[0]) * 1e12 # 排除对角线