summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2020-02-11 17:26:18 +0100
committerJules Laplace <julescarbon@gmail.com>2020-02-11 17:26:18 +0100
commit189d32668383a301473623c41f67e46c98ca502d (patch)
tree18a10bf42d96d39045ee77af2609ac1edef58b5c
parent4711805f5c46ffe80c5227f7ef871214b993f1ec (diff)
save progress
-rw-r--r--cli/app/search/search_dense.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/cli/app/search/search_dense.py b/cli/app/search/search_dense.py
index 5f44995..9c56920 100644
--- a/cli/app/search/search_dense.py
+++ b/cli/app/search/search_dense.py
@@ -135,7 +135,8 @@ def find_dense_embedding_for_images(params, opt_tag="inverse_" + timestamp(), op
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)
new_op_input = tf.Variable(tf.zeros([BATCH_SIZE,] + layer_shape, dtype=tf.float32), name=variable_name, trainable=True)
- raw_op._update_input(op_input_index, new_op_input)
+ maybe_a_tensor = new_op_input + tf.constant(0.0)
+ raw_op._update_input(op_input_index, maybe_a_tensor)
layer_label_variables.append(new_op_input)
# Step counter.