From 44d95c081c9ecbfdfb4a645b40bc1d47cff129b7 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Sun, 23 Feb 2020 15:14:02 +0100 Subject: dont transition everything --- cli/app/search/live.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) (limited to 'cli') 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()))) -- cgit v1.2.3-70-g09d2