From 4e9e37fad40c9a7a20d8c6f8e764d7f9755e5058 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Sat, 22 Feb 2020 17:13:00 +0100 Subject: incorporate layers into live script --- cli/app/search/live.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'cli') diff --git a/cli/app/search/live.py b/cli/app/search/live.py index fb94c12..7dea257 100644 --- a/cli/app/search/live.py +++ b/cli/app/search/live.py @@ -316,6 +316,7 @@ class Interpolator: ] op_input_index = 1 + split_shape = [1, 16] for op_name, layer_id in zip(layer_op_names, range(7)): op_name = 'module_apply_{}/{}'.format(gen_signature, op_name) tensor_name = 'module_apply_{}/Generator_2/split:{}'.format(gen_signature, layer_id + 1) @@ -324,8 +325,8 @@ class Interpolator: ## unlike the encoding, here we update the layer op directly, so we don't need a temporary constant to swap # raw_layer_placeholder = tf.constant(np.zeros(layer_tensor.shape, dtype=np.float32)) - layer_stored = LerpParam('layer_stored_{}'.format(layer_id), shape=layer_tensor.shape, datatype="noise") - layer_stored_mix = LerpParam('layer_stored_mix_{}'.format(layer_id), a_in=layer_tensor, b_in=layer_stored.output, shape=layer_tensor.shape, datatype="noise") + layer_stored = LerpParam('layer_stored_{}'.format(layer_id), shape=split_shape, datatype="noise") + layer_stored_mix = LerpParam('layer_stored_mix_{}'.format(layer_id), a_in=layer_tensor, b_in=layer_stored.output, shape=split_shape, datatype="noise") layer_op._update_input(op_input_index, layer_stored_mix.output) -- cgit v1.2.3-70-g09d2