summaryrefslogtreecommitdiff
path: root/cli/app/search/search_dense.py
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2020-02-11 15:21:33 +0100
committerJules Laplace <julescarbon@gmail.com>2020-02-11 15:21:33 +0100
commitaa6744513ddb38393ddbaaf9703fd644d1da7e99 (patch)
tree9e4d95fbddcd1cb9061c0be39c3e056eb0ab73c4 /cli/app/search/search_dense.py
parentd12aa04e9db12d09ab59d15934d64c185c6535c6 (diff)
just invert latent, also invert all these transformed label layers
Diffstat (limited to 'cli/app/search/search_dense.py')
-rw-r--r--cli/app/search/search_dense.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/cli/app/search/search_dense.py b/cli/app/search/search_dense.py
index 992eb93..4a5bb09 100644
--- a/cli/app/search/search_dense.py
+++ b/cli/app/search/search_dense.py
@@ -132,7 +132,7 @@ def find_dense_embedding_for_images(params, opt_tag="inverse_" + timestamp(), op
for op_name in op_names:
layer_name = 'module_apply_{}/{}'.format(gen_signature, op_name)
variable_name = op_name + "_label"
- raw_op = tf.get_default_graph().get_op_by_name(layer_name)
+ raw_op = tf.get_default_graph().get_operation_by_name(layer_name)
new_op_input = tf.get_variable(name=variable_name, dtype=tf.float32, shape=[BATCH_SIZE,] + layer_shape)
op._update_input(op_input_index, new_op_input)
layer_label_variables.append(new_op_input)