From ab17d875487671655aefc1aab916f5e64132d68e Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Fri, 10 Jan 2020 15:53:14 +0100 Subject: graph magic --- 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 9dcf4a8..2c8c9eb 100644 --- a/cli/app/search/live.py +++ b/cli/app/search/live.py @@ -220,9 +220,9 @@ 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 = tuple([1,] + encoding_latent.get_shape().as_list()[1:]) + encoding_shape_specific = tuple([1,] + encoding_latent.get_shape().as_list()[1:]) encoding_shape = encoding_latent.get_shape().as_list() - encoding_shape_placeholder = tf.constant(np.zeros(encoding_shape, dtype=np.float32)) + encoding_shape_placeholder = tf.constant(np.zeros(encoding_shape_specific, dtype=np.float32)) encoding_stored = LerpParam('encoding_stored', shape=encoding_shape, datatype="encoding") encoding_mix = LerpParam('encoding_mix', a_in=encoding_latent, b_in=encoding_shape_placeholder, shape=encoding_shape, datatype="encoding") # use the placeholder to redirect parts of the graph. -- cgit v1.2.3-70-g09d2