summaryrefslogtreecommitdiff
path: root/cli
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2020-02-11 17:33:31 +0100
committerJules Laplace <julescarbon@gmail.com>2020-02-11 17:33:31 +0100
commitb162b4d010ca80828f7052c9b0f8fab92527f8cd (patch)
tree1cd4e9a5003f7948765b940af75a469cd7e1dfd5 /cli
parent189d32668383a301473623c41f67e46c98ca502d (diff)
save progress
Diffstat (limited to 'cli')
-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 9c56920..0ca5434 100644
--- a/cli/app/search/search_dense.py
+++ b/cli/app/search/search_dense.py
@@ -128,7 +128,7 @@ def find_dense_embedding_for_images(params, opt_tag="inverse_" + timestamp(), op
"Generator_2/concat_6",
]
op_input_index = 1
- layer_shape = [1, 128,]
+ layer_shape = [128,]
for op_name in op_names:
layer_name = 'module_apply_{}/{}'.format(gen_signature, op_name)
variable_name = op_name + "_label"