From f3cb38b9aed7534ff913d868a867161841961707 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Fri, 10 Jan 2020 03:11:14 +0100 Subject: live --- cli/app/search/live.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'cli') 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()))) -- cgit v1.2.3-70-g09d2