Procházet zdrojové kódy

Fix the warning in new PyTorch due to torch.bool

Yichao Zhou před 6 roky
rodič
revize
fcfb25b463
1 změnil soubory, kde provedl 1 přidání a 1 odebrání
  1. 1 1
      lcnn/models/line_vectorizer.py

+ 1 - 1
lcnn/models/line_vectorizer.py

@@ -194,7 +194,7 @@ class LineVectorizer(nn.Module):
             if do_evaluation:
             if do_evaluation:
                 c = (u < v).flatten()
                 c = (u < v).flatten()
             else:
             else:
-                c = torch.zeros_like(label)
+                c = torch.zeros_like(label, dtype=torch.bool)
 
 
                 # sample positive lines
                 # sample positive lines
                 cdx = label.nonzero().flatten()
                 cdx = label.nonzero().flatten()