summaryrefslogtreecommitdiff
path: root/cli/app/search/search_dense.py
diff options
context:
space:
mode:
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 149b8f6..36fc230 100644
--- a/cli/app/search/search_dense.py
+++ b/cli/app/search/search_dense.py
@@ -299,7 +299,7 @@ def find_dense_embedding_for_images(params, opt_tag="inverse_" + timestamp(), op
# --------------------------
if params.decay_lr:
lrate = tf.train.exponential_decay(params.lr, inv_step,
- params.inv_it / params.decay_n, 0.1) #, staircase=True)
+ params.inv_it / params.decay_n, 0.1, staircase=True)
else:
lrate = tf.constant(params.lr)
# trained_params = [label, latent, encoding]