1 files changed, 4 insertions, 4 deletions
diff --git a/cli/app/search/search_dense.py b/cli/app/search/search_dense.py
index fe0c1aa..ac4dc59 100644
--- a/cli/app/search/search_dense.py
+++ b/cli/app/search/search_dense.py
@@ -396,9 +396,9 @@ def find_dense_embedding_for_images(params):
     print("Total iterations: {}".format(params.inv_it))
     for _ in range(params.inv_it):
 
-      _inv_loss, _mse_loss, _feat_loss, _rec_loss, _reg_loss, _dist_loss,\
+      _inv_loss, _mse_loss, _feat_loss,\
           _lrate, _ = sess.run([inv_loss, mse_loss, feat_loss,
-          rec_loss, reg_loss, dist_loss, lrate, inv_train_op])
+          lrate, inv_train_op])
 
       if params.clipping or params.stochastic_clipping:
         sess.run(clip_latent)
@@ -408,9 +408,9 @@ def find_dense_embedding_for_images(params):
         # Log losses.
         etime = time.time() - start_time
         print('It [{:8d}] time [{:5.1f}] total [{:.4f}] mse [{:.4f}] '
-              'feat [{:.4f}] rec [{:.4f}] reg [{:.4f}] dist [{:.4f}] '
+              'feat [{:.4f}] '
               'lr [{:.4f}]'.format(it, etime, _inv_loss, _mse_loss,
-              _feat_loss, _rec_loss, _reg_loss, _dist_loss, _lrate))
+              _feat_loss, _lrate))
 
         sys.stdout.flush()