diff options
| -rw-r--r-- | inversion/live.py | 4 |
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) |