checkpoint

1 files changed, 1 insertions, 1 deletions

diff --git a/main.py b/main.py
index 867c3f8d..9fceccf1 100644
--- a/main.py
+++ b/main.py
@@ -40,7 +40,7 @@ def parse_args():
 def train(model):
     for i in range(hp.num_epochs):
         if i % 100 == 0:
-            fn = f"checkpoint-images/img-epoch{i}"
+            fn = f"checkpoint-images/img-epoch{i}.jpg"
             save_image(fn, tf.squeeze(model.x))
             
         print('batch', i)