From 32545b10ab5b8d3b93567928a423be4dd0351815 Mon Sep 17 00:00:00 2001 From: Benjamin Fiske Date: Sat, 7 May 2022 16:31:49 -0400 Subject: checkpoint --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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) -- cgit v1.2.3-70-g09d2