diff options
Diffstat (limited to 'cli')
| -rw-r--r-- | cli/app/search/search_dense.py | 2 |
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) |