diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2020-01-10 19:57:54 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2020-01-10 19:57:54 +0100 |
| commit | d92efc0d43e566f72fc5bdea9368fdaa0cc5f424 (patch) | |
| tree | 22a21dfb3c607bd0285611a47d4f47cc0d414b63 /cli | |
| parent | 14ec689ccbf85e8761a37e02b63b23fc18fad474 (diff) | |
test label training as part of encoding process
Diffstat (limited to 'cli')
| -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 = {} |