code optimization

caffeine-coder1 · caffeine-coder1 · commit 055a62c8fb0e · 2021-07-01T15:32:48.000+05:30
diff --git a/GAN/WGAN/training.py b/GAN/WGAN/training.py
@@ -139,7 +139,7 @@ def training(opt):
 
                 # ~~~~~~~~~~~~~~~~~~~ loss ~~~~~~~~~~~~~~~~~~~ #
 
-                C_loss = -(torch.mean(real_predict) - torch.mean(fake_predict))
+                C_loss = -(torch.mean(fake_predict) - torch.mean(real_predict))
                 C_loss_avg += C_loss
 
                 # ~~~~~~~~~~~~~~~~~~~ backward ~~~~~~~~~~~~~~~~~~~ #
