type

1 files changed, 9 insertions, 9 deletions

diff --git a/inversion/live.py b/inversion/live.py
index 07af3fe..5bd26fe 100644
--- a/inversion/live.py
+++ b/inversion/live.py
@@ -129,17 +129,17 @@ class Interpolator:
     }
 
   def build(self):
-    # lerp_z = lerp(self.opts, 'latent', [BATCH_SIZE, Z_DIM])
-    # sin_z = sin(self.opts, 'orbit', [BATCH_SIZE, Z_DIM])
-    # lerp_label = lerp(self.opts, 'label', [BATCH_SIZE, N_CLASS])
+    lerp_z = lerp(self.opts, 'latent', [BATCH_SIZE, Z_DIM])
+    sin_z = sin(self.opts, 'orbit', [BATCH_SIZE, Z_DIM])
+    lerp_label = lerp(self.opts, 'label', [BATCH_SIZE, N_CLASS])
 
-    self.opts['z'] = InterpolatorParam(name='z', shape=[BATCH_SIZE, Z_DIM], type='noise')
-    self.opts['y'] = InterpolatorParam(name='y', shape=[BATCH_SIZE, N_CLASS], type='label')
+    # self.opts['z'] = InterpolatorParam(name='z', shape=[BATCH_SIZE, Z_DIM], type='noise')
+    # self.opts['y'] = InterpolatorParam(name='y', shape=[BATCH_SIZE, N_CLASS], type='label')
 
     gen_in = {}
     gen_in['truncation'] = 1.0 # self.opts['truncation'].variable
-    gen_in['z'] = self.opts['z'].variable # lerp_z # + sin_z
-    gen_in['y'] = self.opts['y'].variable
+    gen_in['z'] = lerp_z + sin_z
+    gen_in['y'] = lerp_label
     self.gen_img = generator(gen_in, signature=gen_signature)
 
     # Convert generated image to channels_first.
@@ -169,8 +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)
+    # 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