diff options
| author | nunzip <np.scarh@gmail.com> | 2019-02-27 23:05:02 +0000 |
|---|---|---|
| committer | nunzip <np.scarh@gmail.com> | 2019-02-27 23:05:02 +0000 |
| commit | 6cf023576314b4aa39a8d5d5dc60e656958dde6d (patch) | |
| tree | 052cdad6d37e474afc88c8b49f4e847071087a96 /cgan.py | |
| parent | 367167680e156ac611c5f1db9f9ff7e66d51a8fe (diff) | |
| download | e4-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.py | 12 |
1 files changed, 6 insertions, 6 deletions
@@ -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 |