summaryrefslogtreecommitdiff
path: root/cli/app/search/search_dense.py
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2020-02-11 19:06:37 +0100
committerJules Laplace <julescarbon@gmail.com>2020-02-11 19:06:37 +0100
commit1fc1e2c4922e8d47ff0bcdce08e3e2d1850f3d44 (patch)
tree27e2828c6c88343de57414035a83643a33e58114 /cli/app/search/search_dense.py
parent530bf33a017a59b5d7b987bd665a35abb18d403d (diff)
try g(z) encoding and label embedding
Diffstat (limited to 'cli/app/search/search_dense.py')
-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 458d5b8..4d59da9 100644
--- a/cli/app/search/search_dense.py
+++ b/cli/app/search/search_dense.py
@@ -227,7 +227,7 @@ def find_dense_embedding_for_images(params, opt_tag="inverse_" + timestamp(), op
img_rec_err = params.lambda_mse * img_mse_err + params.lambda_feat * img_feat_err
inv_loss = params.lambda_mse * mse_loss + params.lambda_feat * feat_loss
- inv_loss_quad = params.lambda_mse * mse_loss + params.lambda_feat * feat_loss_quad
+ inv_loss_quad = 5.0 * mse_loss + params.lambda_feat * feat_loss_quad
# --------------------------
# Optimizer.