timing

1 files changed, 3 insertions, 5 deletions

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: