Fix the warning in new PyTorch due to torch.bool

zhou13 · Oct 17, 2019 · fcfb25b · fcfb25b
1 parent 2c9d414
commit fcfb25b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lcnn/models/line_vectorizer.py b/lcnn/models/line_vectorizer.py
@@ -194,7 +194,7 @@ def sample_lines(self, meta, jmap, joff, do_evaluation):
             if do_evaluation:
                 c = (u < v).flatten()
             else:
-                c = torch.zeros_like(label)
+                c = torch.zeros_like(label, dtype=torch.bool)
 
                 # sample positive lines
                 cdx = label.nonzero().flatten()