moreo
/
kernel_authorship
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

adding pos and neg losses

This commit is contained in:
Alejandro Moreo Fernandez 2021-02-04 18:52:46 +01:00
parent 1385d2ac95
commit 38867371fa
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/model/classifiers.py
View File

@ -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)