diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2020-02-11 19:12:19 +0100 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2020-02-11 19:12:19 +0100 |
| commit | b39078be6f4711a07a1a13d317312863dc26e909 (patch) | |
| tree | 31833f281f8263b54a551ccdc84ce74408304032 /cli/app/search | |
| parent | ec9421c4d52e92d9a2cb3e8d5edb019e0d4e5f8d (diff) | |
try g(z) encoding and label embedding
Diffstat (limited to 'cli/app/search')
| -rw-r--r-- | cli/app/search/search_dense.py | 3 |
1 files changed, 2 insertions, 1 deletions
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] |