diff options
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 63357c7..b2aa7c3 100644 --- a/cli/app/search/live.py +++ b/cli/app/search/live.py @@ -220,10 +220,10 @@ class Interpolator: gen_layer_name = 'module_apply_' + gen_signature + '/' + params.inv_layer encoding_latent = tf.get_default_graph().get_tensor_by_name(gen_layer_name) - encoding_shape = [1,] + gen_encoding.get_shape().as_list()[1:] + encoding_shape = [1,] + encoding_latent.get_shape().as_list()[1:] encoding_stored = LerpParam('encoding_stored', shape=encoding_shape, datatype="encoding") encoding_mix = LerpParam('encoding_mix', a_in=encoding_latent, b_in=encoding_stored, shape=encoding_shape, datatype="encoding") - tf.contrib.graph_editor.swap_ts(gen_encoding, encoding_mix) + tf.contrib.graph_editor.swap_ts(encoding_latent, encoding_mix) sys.stderr.write("Sin params: {}\n".format(", ".join(self.sin_params.keys()))) sys.stderr.write("Lerp params: {}\n".format(", ".join(self.lerp_params.keys()))) |