aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--cgan.py12
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