minor fix gan

yusugomori · Mar 26, 2019 · 1c1e448 · 1c1e448
1 parent 78df046
commit 1c1e448
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/gan_fashion_mnist.py b/models/gan_fashion_mnist.py
@@ -226,4 +226,4 @@ def gen_noise(batch_size):
             plt.tight_layout()
             # plt.show()
             template = '{}/gan_fashion_mnist_epoch_{:0>4}.png'
-            plt.savefig(template.format(out_path, epoch), dpi=300)
+            plt.savefig(template.format(out_path, epoch+1), dpi=300)