From 4711805f5c46ffe80c5227f7ef871214b993f1ec Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Tue, 11 Feb 2020 16:49:42 +0100 Subject: save progress --- cli/app/search/search_dense.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'cli') diff --git a/cli/app/search/search_dense.py b/cli/app/search/search_dense.py index 9e694d8..5f44995 100644 --- a/cli/app/search/search_dense.py +++ b/cli/app/search/search_dense.py @@ -133,7 +133,8 @@ def find_dense_embedding_for_images(params, opt_tag="inverse_" + timestamp(), op layer_name = 'module_apply_{}/{}'.format(gen_signature, op_name) 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) + # 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) layer_label_variables.append(new_op_input) -- cgit v1.2.3-70-g09d2