mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update loss with lr crossgan

Browse Source
This commit is contained in:
James Betker 2020-08-26 17:57:22 -06:00
parent 80aa83bfd2
commit 25832930db
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
codes/models/loss.py
View File

@ -26,7 +26,7 @@ class GANLoss(nn.Module):
self.real_label_val = real_label_val
self.fake_label_val = fake_label_val
if self.gan_type in ['gan', 'ragan', 'pixgan', 'pixgan_fea', 'crossgan']:
if self.gan_type in ['gan', 'ragan', 'pixgan', 'pixgan_fea', 'crossgan', 'crossgan_lrref']:
self.loss = nn.BCEWithLogitsLoss()
elif self.gan_type == 'lsgan':
self.loss = nn.MSELoss()
@ -40,7 +40,7 @@ class GANLoss(nn.Module):
return torch.empty_like(input).fill_(self.fake_label_val)
def forward(self, input, target_is_real):
if self.gan_type in ['pixgan', 'pixgan_fea', 'crossgan'] and not isinstance(target_is_real, bool):
if self.gan_type in ['pixgan', 'pixgan_fea', 'crossgan', 'crossgan_lrref'] and not isinstance(target_is_real, bool):
target_label = target_is_real
else:
target_label = self.get_target_label(input, target_is_real)