Update vae_shi.py

yoyo-yun · Mar 24, 2018 · 898430c · 898430c
1 parent bf0a5ff
commit 898430c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/vae_shi.py b/vae_shi.py
@@ -136,3 +136,6 @@ def on_epoch_end(self, epoch, logs=None):
         callbacks=[evaluator])
 
 vae.save_weights('shi.model')
+
+for i in range(20):
+    print gen()