diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2020-01-10 15:30:34 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2020-01-10 15:30:34 +0100 |
| commit | 5852035247313594ad6ca240ccf52eee56832d7c (patch) | |
| tree | eee3e161b0b118d4821a7fad196539c5995077fa | |
| parent | 4ad173ed0f8cc623549d2ccbc21114ac28237c4b (diff) | |
graph magic
| -rw-r--r-- | cli/app/search/json.py | 2 | ||||
| -rw-r--r-- | cli/app/search/live.py | 4 |
2 files changed, 3 insertions, 3 deletions
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()))) |