type

1 files changed, 2 insertions, 2 deletions

diff --git a/inversion/live.py b/inversion/live.py
index 3e40be5..dad916a 100644
--- a/inversion/live.py
+++ b/inversion/live.py
@@ -169,6 +169,8 @@ class Interpolator:
       sys.stderr.write('{} not a valid option\n'.format(key))
 
   def on_step(self, i, sess):
+    self.opts['z'].randomize(truncation=1.0)
+    self.opts['y'].randomize(num_classes=1.0)
     gen_images = sess.run(self.gen_img, feed_dict=self.get_feed_dict())
     return gen_images
 
@@ -212,8 +214,6 @@ class Listener:
     for i in range(99999):
       if (i % 100) == 0:
         print("Step {}".format(i))
-      self.opts['z'].randomize(truncation=1.0)
-      self.opts['y'].randomize(num_classes=1.0)
       gen_time = time.time()
       self.interpolator.opts['time'].assign(gen_time)
       gen_images = self.interpolator.on_step(i, self.sess)