Skip to content

Commit 770b653

Browse files
authored
Update losses.py
1 parent ff97df4 commit 770b653

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/utils/losses.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -153,7 +153,7 @@ def forward(self, inst_embed, proxy, negative_mask, labels, temperature, margin)
153153

154154
class Conditional_Contrastive_loss_plus(torch.nn.Module):
155155
def __init__(self, device, batch_size, pos_collected_numerator):
156-
super(Conditional_Contrastive_loss, self).__init__()
156+
super(Conditional_Contrastive_loss_plus, self).__init__()
157157
self.device = device
158158
self.batch_size = batch_size
159159
self.pos_collected_numerator = pos_collected_numerator

0 commit comments

Comments
 (0)