msight-tech / research-xbm

XBM: Cross-Batch Memory for Embedding Learning
Other
304 stars 38 forks source link

hard_mining #15

Open wangyf8848 opened 1 year ago

wangyf8848 commented 1 year ago

请问在ms损失中,hard_mining时调用torch.min(pospair)时,pos_pair_有时长度为0报错怎么办

wangxinrugithub commented 1 year ago

请问在ms损失中,hard_mining时调用torch.min(pospair)时,pos_pair_有时长度为0报错怎么办

我也遇到了这个问题,我加了一个判断,如果pos_pair_为0就continue。 但我现在遇到了一个新问题,loss不收敛