From 5948a231b89e4dfd6acfd35de821fbe085ec4114 Mon Sep 17 00:00:00 2001 From: Benjamin Fiske Date: Sat, 7 May 2022 16:23:24 -0400 Subject: saving checkpoints --- main.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/main.py b/main.py index 45714611..87f36013 100644 --- a/main.py +++ b/main.py @@ -39,9 +39,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}" + save_image(fn, 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() -- cgit v1.2.3-70-g09d2 From cf704e000e901ee911fd4b9527bb745d13d345b6 Mon Sep 17 00:00:00 2001 From: Benjamin Fiske Date: Sat, 7 May 2022 16:30:57 -0400 Subject: tf.squeeze --- __pycache__/hyperparameters.cpython-38.pyc | Bin 324 -> 369 bytes __pycache__/losses.cpython-38.pyc | Bin 4601 -> 4646 bytes main.py | 2 +- 3 files changed, 1 insertion(+), 1 deletion(-) diff --git a/__pycache__/hyperparameters.cpython-38.pyc b/__pycache__/hyperparameters.cpython-38.pyc index e42d48d1..5fc87730 100644 Binary files a/__pycache__/hyperparameters.cpython-38.pyc and b/__pycache__/hyperparameters.cpython-38.pyc differ diff --git a/__pycache__/losses.cpython-38.pyc b/__pycache__/losses.cpython-38.pyc index 07bf7a33..703088c9 100644 Binary files a/__pycache__/losses.cpython-38.pyc and b/__pycache__/losses.cpython-38.pyc differ diff --git a/main.py b/main.py index 87f36013..867c3f8d 100644 --- a/main.py +++ b/main.py @@ -41,7 +41,7 @@ def train(model): for i in range(hp.num_epochs): if i % 100 == 0: fn = f"checkpoint-images/img-epoch{i}" - save_image(fn, model.x) + save_image(fn, tf.squeeze(model.x)) print('batch', i) model.train_step() -- cgit v1.2.3-70-g09d2 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