1 files changed, 6 insertions, 6 deletions
diff --git a/cgan.py b/cgan.py
index fa05311..3b3d0b7 100644
--- a/cgan.py
+++ b/cgan.py
@@ -164,12 +164,12 @@ class CGAN():
             if epoch % sample_interval == 0:
                 self.sample_images(epoch)
             
-            if graph:
-              plt.plot(xaxis,loss[0])
-              plt.plot(xaxis,loss[1])
-              plt.legend(('Discriminator', 'Generator'), loc='best')
-              plt.xlabel('Epoch')
-              plt.ylabel('Binary Crossentropy Loss')
+        if graph:
+          plt.plot(xaxis,loss[0])
+          plt.plot(xaxis,loss[1])
+          plt.legend(('Discriminator', 'Generator'), loc='best')
+          plt.xlabel('Epoch')
+          plt.ylabel('Binary Crossentropy Loss')
 
     def sample_images(self, epoch):
         r, c = 2, 5