Reduce memory usage during evaluation

zhou13 · Oct 17, 2019 · 2c9d414 · 2c9d414
1 parent 7fdb112
commit 2c9d414
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lcnn/models/line_vectorizer.py b/lcnn/models/line_vectorizer.py
@@ -159,7 +159,7 @@ def sample_lines(self, meta, jmap, joff, do_evaluation):
             max_K = M.n_dyn_junc // n_type
             N = len(junc)
             if do_evaluation:
-                K = min(int((jmap > M.eval_junc_thres).float().sum().item()), 2 * max_K)
+                K = min(int((jmap > M.eval_junc_thres).float().sum().item()), max_K)
             else:
                 K = min(int(N * 2 + 2), max_K)
             device = jmap.device