Use tensor.bool instead of tensor.byte

V2AI · Jan 6, 2020 · b567905 · b567905
1 parent fa13f93
commit b567905
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/det3d/models/bbox_heads/mg_head.py b/det3d/models/bbox_heads/mg_head.py
@@ -1035,7 +1035,7 @@ def get_task_detections(
                     dir_labels = selected_dir_labels
                     opp_labels = (
                         (box_preds[..., -1] - self.direction_offset) > 0
-                    ) ^ dir_labels.byte()
+                    ) ^ dir_labels.bool()
                     box_preds[..., -1] += torch.where(
                         opp_labels,
                         torch.tensor(np.pi).type_as(box_preds),