bugfix

Andrewzh112 · Dec 14, 2020 · c7d4a19 · c7d4a19
1 parent cd3b55e
commit c7d4a19
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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):