deviandice/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix? cosine loss

Browse Source
This commit is contained in:
James Betker 2020-09-22 18:18:35 -06:00
parent 2e18c4c22d
commit 7b60d9e0d8
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/models/steps/losses.py
View File

@ -82,7 +82,7 @@ class FeatureLoss(ConfigurableLoss):
logits_real = self.netF(state[self.opt['real']])
logits_fake = self.netF(state[self.opt['fake']])
if self.opt['criterion'] == 'cosine':
return self.criterion(logits_fake, logits_real, 1)
return self.criterion(logits_fake, logits_real, torch.ones_like(logits_fake))
else:
return self.criterion(logits_fake, logits_real)