diff options
Diffstat (limited to 'cli')
| -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] |