diff options
| -rw-r--r-- | cli/app/search/live.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/cli/app/search/live.py b/cli/app/search/live.py index 183bab9..24d49e6 100644 --- a/cli/app/search/live.py +++ b/cli/app/search/live.py @@ -212,14 +212,14 @@ class Interpolator: # Latent - stored vector latent_stored = LerpParam(name='latent_stored', shape=[BATCH_SIZE, Z_DIM], datatype="noise") - latent_stored_mix = LerpParam('latent_stored_mix', a_in=z_mix.output, b_in=latent_stored.variable, shape=[BATCH_SIZE, Z_DIM], datatype="input") + latent_stored_mix = LerpParam('latent_stored_mix', a_in=z_mix.output, b_in=latent_stored.output, shape=[BATCH_SIZE, Z_DIM], datatype="input") # Label lerp_label = LerpParam('label', shape=[BATCH_SIZE, N_CLASS], datatype="label") # Latent - stored vector label_stored = LerpParam(name='label_stored', shape=[BATCH_SIZE, N_CLASS], datatype="label") - label_stored_mix = LerpParam('label_stored_mix', a_in=z_mix.output, b_in=latent_stored.variable, shape=[BATCH_SIZE, Z_DIM], datatype="input") + label_stored_mix = LerpParam('label_stored_mix', a_in=lerp_label.output, b_in=label_stored.output, shape=[BATCH_SIZE, Z_DIM], datatype="input") # Generator gen_in = {} |