From aaece2d44b6dd97658d74860e0bc5e777e59a6f1 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Sat, 21 Dec 2019 22:32:01 +0100 Subject: timing --- inversion/live.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) (limited to 'inversion') diff --git a/inversion/live.py b/inversion/live.py index 50ac25f..9e9c5c3 100644 --- a/inversion/live.py +++ b/inversion/live.py @@ -290,17 +290,15 @@ class Listener: if i == 0: print("Loading network") elif i == 1: - print("Processing") - elif (i % 100) == 0: - print("Step {}".format(i)) + print("Processing!") gen_time = time.time() gen_images = interpolator.on_step(i, dt, self.sess) if gen_images is None: print("Exiting...") break if (i % 100) == 0 or i == 1: - print(gen_images.shape) - print("Generation time: {:.1f}s".format(time.time() - gen_time)) + # print(gen_images.shape) + print("Step {}. Generation time: {:.2f}s".format(i, time.time() - gen_time)) out_img = vs.data2pil(gen_images[0]) if out_img is not None: #if out_img.resize_before_sending: -- cgit v1.2.3-70-g09d2