diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2020-02-11 15:28:30 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2020-02-11 15:28:30 +0100 |
| commit | c5b066ebec6a070fd9ac110ec8f2c22028b2d98a (patch) | |
| tree | 2c4960a7b53fafcd6d20680f97bdc3bcee3bb21f /cli/app/search/search_dense.py | |
| parent | e29787e034ce7131d68ea2c14ac3db45fdef57f8 (diff) | |
save progress
Diffstat (limited to 'cli/app/search/search_dense.py')
| -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 4a5bb09..9e694d8 100644 --- a/cli/app/search/search_dense.py +++ b/cli/app/search/search_dense.py @@ -134,7 +134,7 @@ def find_dense_embedding_for_images(params, opt_tag="inverse_" + timestamp(), op variable_name = op_name + "_label" 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) + raw_op._update_input(op_input_index, new_op_input) layer_label_variables.append(new_op_input) # Step counter. |