From 6cf023576314b4aa39a8d5d5dc60e656958dde6d Mon Sep 17 00:00:00 2001 From: nunzip Date: Wed, 27 Feb 2019 23:05:02 +0000 Subject: Fix graph function --- cgan.py | 12 ++++++------ 1 file 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 -- cgit v1.2.3-54-g00ecf