From cf704e000e901ee911fd4b9527bb745d13d345b6 Mon Sep 17 00:00:00 2001 From: Benjamin Fiske Date: Sat, 7 May 2022 16:30:57 -0400 Subject: tf.squeeze --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'main.py') 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