summaryrefslogtreecommitdiff
path: root/cli
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2020-02-11 19:10:10 +0100
committerJules Laplace <julescarbon@gmail.com>2020-02-11 19:10:10 +0100
commitc22b4f180a879fffe6c163b99676f65cfcc49cd4 (patch)
tree3b1c716b97df975bf3cbbe9ab26f12b2eca29440 /cli
parent1fc1e2c4922e8d47ff0bcdce08e3e2d1850f3d44 (diff)
try g(z) encoding and label embedding
Diffstat (limited to 'cli')
-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 4d59da9..458d5b8 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 = 5.0 * mse_loss + params.lambda_feat * feat_loss_quad
+ inv_loss_quad = params.lambda_mse * mse_loss + params.lambda_feat * feat_loss_quad
# --------------------------
# Optimizer.