diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2020-01-10 03:11:14 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2020-01-10 03:11:14 +0100 |
| commit | f3cb38b9aed7534ff913d868a867161841961707 (patch) | |
| tree | 47b53e33b96147a6cb14dc71f5e81771767df8ae | |
| parent | 42178b4350d6b2d99a69fccd238863cb9dd1f327 (diff) | |
live
| -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()))) |