From a204f0af70c62f23f57d13f99565b036ae167f03 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Wed, 19 Feb 2020 02:28:46 +0100 Subject: reverse label also --- cli/app/search/search_dense.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'cli') diff --git a/cli/app/search/search_dense.py b/cli/app/search/search_dense.py index b2c77f4..00ae5b3 100644 --- a/cli/app/search/search_dense.py +++ b/cli/app/search/search_dense.py @@ -330,7 +330,7 @@ def find_dense_embedding_for_images(params, opt_tag="inverse_" + timestamp(), op if params.inv_layer == 'latent': trained_params = [latent] else: - trained_params = [latent, encoding] + trained_params = [label, latent, encoding] if params.invert_labels: trained_params += layer_label_variables -- cgit v1.2.3-70-g09d2