From b39078be6f4711a07a1a13d317312863dc26e909 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Tue, 11 Feb 2020 19:12:19 +0100 Subject: try g(z) encoding and label embedding --- 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 f065f56..75bba19 100644 --- a/cli/app/search/search_dense.py +++ b/cli/app/search/search_dense.py @@ -430,7 +430,8 @@ def find_dense_embedding_for_images(params, opt_tag="inverse_" + timestamp(), op out_data['layer_labels'].append(layer[i]) write_pickle(out_data, fp_out_pkl) out_lat[out_i] = latent_trained[i] - out_enc[out_i] = enc_trained[i] + if params.inv_layer != 'latent': + out_enc[out_i] = enc_trained[i] out_images[out_i] = image_batch[i] out_labels[out_i] = label_trained[i] -- cgit v1.2.3-70-g09d2