Skip to content

Commit 21e9032

Browse files
committed
bugfix
1 parent 2b0af8d commit 21e9032

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

angle_emb/angle.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -741,7 +741,7 @@ def __call__(self,
741741
elif self.dataset_format == DatasetFormats.C:
742742
text = outputs[::2]
743743
positive = outputs[1::2]
744-
loss = contrastive_with_negative_loss(text, positive, negative=None, tau=self.ibn_tau)
744+
loss = contrastive_with_negative_loss(text, positive, neg=None, tau=self.ibn_tau)
745745
else:
746746
raise NotImplementedError
747747
return loss

0 commit comments

Comments
 (0)