From c7d4a19a83f0dd559b6a8a5a2cd79891c600240b Mon Sep 17 00:00:00 2001 From: Andrew Zhao Date: Mon, 14 Dec 2020 11:23:51 +0800 Subject: [PATCH] bugfix --- sagan/loss.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sagan/loss.py b/sagan/loss.py index 63f2067..7c8d5c4 100644 --- a/sagan/loss.py +++ b/sagan/loss.py @@ -46,7 +46,7 @@ def forward(self, fake_logits, mode, real_logits=None): def _generator_loss(self, fake_logits): return - fake_logits.mean() - def __discriminator_loss(self, real_logits, fake_logits): + def _discriminator_loss(self, real_logits, fake_logits): return - real_logits.mean() + fake_logits.mean() def get_interpolates(self, reals, fakes):