version

1 files changed, 8 insertions, 5 deletions

diff --git a/inversion/live.py b/inversion/live.py
index afeeccb..7f07189 100644
--- a/inversion/live.py
+++ b/inversion/live.py
@@ -129,14 +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])
+    self.opts['y'] = InterpolatorParam(name='y', shape=[BATCH_SIZE, N_CLASS])
 
     gen_in = {}
     gen_in['truncation'] = 1.0 # self.opts['truncation'].variable
-    gen_in['z'] = lerp_z # + sin_z
-    gen_in['y'] = lerp_label
+    gen_in['z'] = self.opts['z'].variable # lerp_z # + sin_z
+    gen_in['y'] = self.opts['y'].variable
     gen_img = generator(gen_in, signature=gen_signature)
 
     # Convert generated image to channels_first.