diff options
| -rw-r--r-- | cli/app/search/live.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/cli/app/search/live.py b/cli/app/search/live.py index b2aa7c3..3d884e1 100644 --- a/cli/app/search/live.py +++ b/cli/app/search/live.py @@ -221,9 +221,10 @@ class Interpolator: encoding_latent = tf.get_default_graph().get_tensor_by_name(gen_layer_name) encoding_shape = [1,] + encoding_latent.get_shape().as_list()[1:] + print(encoding_shape) 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(encoding_latent, encoding_mix) + encoding_mix = LerpParam('encoding_mix', a_in=encoding_latent, b_in=encoding_stored.output, shape=encoding_shape, datatype="encoding") + tf.contrib.graph_editor.swap_ts(encoding_latent, encoding_mix.output) sys.stderr.write("Sin params: {}\n".format(", ".join(self.sin_params.keys()))) sys.stderr.write("Lerp params: {}\n".format(", ".join(self.lerp_params.keys()))) |