bojone / SimCSE

SimCSE在中文任务上的简单实验
591 stars 83 forks source link

运行环境同 readme, simcse_loss 中 `K.eye(K.shape(y_pred)[0])` 在 `encoder.compile` 时直接卡死 #1

Closed ericxsun closed 3 years ago

bojone commented 3 years ago

你试试 K.eye(K.shape(y_pred)[0]) 改为 tf.eye(K.shape(y_pred)[0]),我昨天试了一个环境也在这句报错了~

ericxsun commented 3 years ago

你试试 K.eye(K.shape(y_pred)[0]) 改为 tf.eye(K.shape(y_pred)[0]),我昨天试了一个环境也在这句报错了~

非常感谢,可以正确运行了。