summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2020-02-22 17:18:07 +0100
committerJules Laplace <julescarbon@gmail.com>2020-02-22 17:18:07 +0100
commit4507a1538b8af43ca8c5f22306b19d13a15ad542 (patch)
tree8b5ab1efd363d54afb12cc9df3698af7b8283a46
parentb2f9be92ecdd92e59a3a97ed13dad3d1b317fe0b (diff)
incorporate layers into live script
-rw-r--r--cli/app/search/live.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/cli/app/search/live.py b/cli/app/search/live.py
index a003350..c9bec1a 100644
--- a/cli/app/search/live.py
+++ b/cli/app/search/live.py
@@ -316,12 +316,12 @@ class Interpolator:
]
op_input_index = 1
- split_shape = [1, 16]
+ split_shape = [1, 128]
for op_name, layer_id in zip(layer_op_names, range(7)):
tensor_name = 'module_apply_{}/Generator_2/split:{}'.format(gen_signature, layer_id + 1)
layer_tensor = tf.get_default_graph().get_tensor_by_name(tensor_name)
- op_name = 'module_apply_{}/{}'.format(gen_signature, op_name)
+x op_name = 'module_apply_{}/{}'.format(gen_signature, op_name)
layer_op = tf.get_default_graph().get_operation_by_name(op_name)
## unlike the encoding, here we update the layer op directly, so we don't need a temporary constant to swap