From 5852035247313594ad6ca240ccf52eee56832d7c Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Fri, 10 Jan 2020 15:30:34 +0100 Subject: graph magic --- cli/app/search/json.py | 2 +- cli/app/search/live.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) (limited to 'cli') diff --git a/cli/app/search/json.py b/cli/app/search/json.py index ced84df..ea70fd6 100644 --- a/cli/app/search/json.py +++ b/cli/app/search/json.py @@ -17,7 +17,7 @@ def make_params_latent(tag): "inv_layer": "latent", "decay_lr": True, "inv_it": 15000, - "generator_path": "https://tfhub.dev/deepmind/biggan-512/2", + "generator_path": "https://tfhub.dev/deepmind/biggan-512/2", "attention_map_layer": "Generator_2/attention/Softmax:0", "pre_trained_latent": False, "lambda_dist": 0.0, diff --git a/cli/app/search/live.py b/cli/app/search/live.py index 63799a1..365e2c6 100644 --- a/cli/app/search/live.py +++ b/cli/app/search/live.py @@ -229,8 +229,8 @@ class Interpolator: # use the placeholder to redirect parts of the graph. # - computed encoding goes into the encoding_mix # - encoding mix output goes into the main biggan graph - tf.contrib.graph_editor.reroute_ts(encoding_latent, encoding_shape_placeholder) - tf.contrib.graph_editor.reroute_ts(encoding_mix.output, encoding_shape_placeholder) + tf.contrib.graph_editor.swap_ts(encoding_latent, encoding_shape_placeholder) + tf.contrib.graph_editor.swap_ts(encoding_shape_placeholder, encoding_mix.output) 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