aboutsummaryrefslogtreecommitdiff
path: root/cgan.py
diff options
context:
space:
mode:
authornunzip <np.scarh@gmail.com>2019-02-27 23:05:02 +0000
committernunzip <np.scarh@gmail.com>2019-02-27 23:05:02 +0000
commit6cf023576314b4aa39a8d5d5dc60e656958dde6d (patch)
tree052cdad6d37e474afc88c8b49f4e847071087a96 /cgan.py
parent367167680e156ac611c5f1db9f9ff7e66d51a8fe (diff)
downloade4-gan-6cf023576314b4aa39a8d5d5dc60e656958dde6d.tar.gz
e4-gan-6cf023576314b4aa39a8d5d5dc60e656958dde6d.tar.bz2
e4-gan-6cf023576314b4aa39a8d5d5dc60e656958dde6d.zip
Fix graph function
Diffstat (limited to 'cgan.py')
-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