stage1的compute_loss_of_similarity是不是有问题

ZhuiyiTechnology / roformer-sim

SimBERT升级版（SimBERTv2）！

Apache License 2.0

439 stars 73 forks source link

Open lxiangge1126 opened 3 years ago

lxiangge1126 commented 3 years ago

大佬，loss中的compute_loss_of_similarity方法，return loss是不是应该换成return K.mean(loss)

bojone commented 3 years ago

没报错的话应该是等价的