summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2020-01-10 15:53:14 +0100
committerJules Laplace <julescarbon@gmail.com>2020-01-10 15:53:14 +0100
commitab17d875487671655aefc1aab916f5e64132d68e (patch)
treeae65eaef0e65bd6d247d1ada661d53d10b605f99
parent1b9eec18c423c881abee0ea6ab5048be92055acc (diff)
graph magic
-rw-r--r--cli/app/search/live.py4
1 files changed, 2 insertions, 2 deletions
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.