From a51daacde24b2fb6f311f736237f1ac91a2ca2af Mon Sep 17 00:00:00 2001 From: James Betker Date: Mon, 2 Nov 2020 08:45:46 -0700 Subject: [PATCH] Fix reporting of d_fake_diff for generators --- codes/models/steps/losses.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/codes/models/steps/losses.py b/codes/models/steps/losses.py index 5ea6c113..d7efdfb6 100644 --- a/codes/models/steps/losses.py +++ b/codes/models/steps/losses.py @@ -176,10 +176,10 @@ class GeneratorGanLoss(ConfigurableLoss): if self.detach_real: pred_d_real = pred_d_real.detach() pred_g_fake = netD(*fake) - d_fake_diff = self.criterion(pred_g_fake - torch.mean(pred_d_real), True) + d_fake_diff = pred_g_fake - torch.mean(pred_d_real) self.metrics.append(("d_fake_diff", torch.mean(d_fake_diff))) loss = (self.criterion(pred_d_real - torch.mean(pred_g_fake), False) + - d_fake_diff) / 2 + self.criterion(d_fake_diff, True)) / 2 else: raise NotImplementedError if self.min_loss != 0: