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