diff options
Diffstat (limited to 'inversion')
| -rw-r--r-- | inversion/image_inversion_inception.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/inversion/image_inversion_inception.py b/inversion/image_inversion_inception.py index c4eb5b5..92a472a 100644 --- a/inversion/image_inversion_inception.py +++ b/inversion/image_inversion_inception.py @@ -467,7 +467,9 @@ for image_batch, label_batch in image_gen: # Save target images and reconstructions. if params.save_progress: assert SAMPLE_SIZE <= BATCH_SIZE + gen_time = time.time() gen_images = sess.run(gen_img) + print("Generation time: {:.1f}s".format(time.time() - gen_time)) inv_batch = vs.interleave(vs.data2img(image_batch[BATCH_SIZE - SAMPLE_SIZE:]), vs.data2img(gen_images[BATCH_SIZE - SAMPLE_SIZE:])) inv_batch = vs.grid_transform(inv_batch) |