merge dev2

1 files changed, 10 insertions, 0 deletions

diff --git a/main.py b/main.py
index 605ddee7..4497cce0 100644
--- a/main.py
+++ b/main.py
@@ -40,9 +40,17 @@ def parse_args():
 
 def train(model):
     for i in range(hp.num_epochs):
+        if i % 100 == 0:
+            fn = f"checkpoint-images/img-epoch{i}.jpg"
+            save_image(fn, tf.squeeze(model.x))
+            
         print('batch', i)
         model.train_step()
 
+def save_image(filename, image):
+    imsave(filename, image)
+
+
 def main():
     """ Main function. """
     if os.path.exists(ARGS.content):
@@ -66,6 +74,8 @@ def main():
 
     imsave(ARGS.savefile, final_image)
 
+    
+
 
 ARGS = parse_args()
 main()