diff options
Diffstat (limited to 'cli/app/search/live.py')
| -rw-r--r-- | cli/app/search/live.py | 11 |
1 files changed, 8 insertions, 3 deletions
diff --git a/cli/app/search/live.py b/cli/app/search/live.py index b680f47..22441e2 100644 --- a/cli/app/search/live.py +++ b/cli/app/search/live.py @@ -302,8 +302,8 @@ class Interpolator: # Make all the stored lerps use the same interpolation amount. tf.contrib.graph_editor.reroute_ts(encoding_stored.n.variable, latent_stored.n.variable) tf.contrib.graph_editor.reroute_ts(encoding_stored.n.variable, label_stored.n.variable) - tf.contrib.graph_editor.reroute_ts(encoding_stored_mix.n.variable, latent_stored_mix.n.variable) - tf.contrib.graph_editor.reroute_ts(encoding_stored_mix.n.variable, label_stored_mix.n.variable) + # tf.contrib.graph_editor.reroute_ts(encoding_stored_mix.n.variable, latent_stored_mix.n.variable) + # tf.contrib.graph_editor.reroute_ts(encoding_stored_mix.n.variable, label_stored_mix.n.variable) layer_op_names = [ "Generator_2/concat", @@ -316,6 +316,8 @@ class Interpolator: ] op_input_index = 1 + # first_mix = encoding_stored_mix + first_mix = None # split_shape = [1, 128] for op_name, layer_id in zip(layer_op_names, range(7)): tensor_name = 'module_apply_{}/linear_1/MatMul:0'.format(gen_signature) @@ -332,7 +334,10 @@ class Interpolator: layer_op._update_input(op_input_index, layer_stored_mix.output) # Make all the stored lerps use the same interpolation amount. - tf.contrib.graph_editor.reroute_ts(encoding_stored_mix.n.variable, layer_stored_mix.n.variable) + if first_mix is None: + first_mix = layer_stored_mix + else: + tf.contrib.graph_editor.reroute_ts(first_mix.n.variable, layer_stored_mix.n.variable) sys.stderr.write("Sin params: {}\n".format(", ".join(self.sin_params.keys()))) sys.stderr.write("Lerp params: {}\n".format(", ".join(self.lerp_params.keys()))) |