From 38867371fabd40955cbd0ffd3a1abf15fad9397b Mon Sep 17 00:00:00 2001 From: Alex Moreo Date: Thu, 4 Feb 2021 18:52:46 +0100 Subject: [PATCH] adding pos and neg losses --- src/model/classifiers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/model/classifiers.py b/src/model/classifiers.py index f23d826..b6a788c 100644 --- a/src/model/classifiers.py +++ b/src/model/classifiers.py @@ -177,7 +177,7 @@ class AuthorshipAttributionClassifier(nn.Module): phi = self.projector(xi) contrastive_loss, neg_loss, pos_loss = criterion(phi, torch.as_tensor(yi).to(self.device)) #contrastive_loss = neg_loss + pos_loss - losses.append(contrastive_loss.item()) + losses.append((neg_loss + pos_loss).item()) neg_losses_val.append(neg_loss.item()) pos_losses_val.append(pos_loss.item()) val_loss = np.mean(losses)