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

update in the frobenius norm, adding and squrt

This commit is contained in:
Alejandro Moreo Fernandez 2021-02-09 15:38:18 +01:00
parent 38867371fa
commit 9f3567c4f8
1 changed files with 2 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/losses.py
View File

@ -107,16 +107,6 @@ class SupConLoss1View(nn.Module):
self.base_temperature = base_temperature
def forward(self, features, labels):
"""Compute loss for model. If both `labels` and `mask` are None,
it degenerates to SimCLR unsupervised loss:
https://arxiv.org/pdf/2002.05709.pdf
Args:
features: hidden vector of shape [bsz, ndim].
labels: ground truth of shape [bsz].
Returns:
A loss scalar.
"""
device = (torch.device('cuda')
if features.is_cuda
else torch.device('cpu'))
@ -146,7 +136,8 @@ class SupConLoss1View(nn.Module):
#return neg_loss, pos_loss
#balanced_loss = pos_loss + neg_loss
#return balanced_loss
return torch.mean((cross_upper-mask_upper)**2), neg_loss, pos_loss
n=len(mask_upper)
return (1/n)*torch.sqrt(torch.sum((cross_upper-mask_upper)**2)), neg_loss, pos_loss
def mse(input, target, label):