1 files changed, 2 insertions, 1 deletions
diff --git a/cgan.py b/cgan.py
index 880a8b8..ebdfab2 100755
--- a/cgan.py
+++ b/cgan.py
@@ -142,9 +142,10 @@ class CGAN():
             # Train the discriminator
             if smooth == True:
                 d_loss_real = self.discriminator.train_on_batch([imgs, labels], valid*0.9)
+                d_loss_fake = self.discriminator.train_on_batch([gen_imgs, labels], valid*0.1)
             else:
                 d_loss_real = self.discriminator.train_on_batch([imgs, labels], valid)
-            d_loss_fake = self.discriminator.train_on_batch([gen_imgs, labels], fake)
+                d_loss_fake = self.discriminator.train_on_batch([gen_imgs, labels], fake)
             d_loss = 0.5 * np.add(d_loss_real, d_loss_fake)
 
             # ---------------------