Merge remote-tracking branch 'upstream/main'

LostSunset · Oct 13, 2024 · 319ec57 · 319ec57
2 parents ebd9599 + 28fd05e
commit 319ec57
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/detectron2/structures/rotated_boxes.py b/detectron2/structures/rotated_boxes.py
@@ -229,9 +229,9 @@ def clone(self) -> "RotatedBoxes":
         """
         return RotatedBoxes(self.tensor.clone())
 
-    def to(self, device: torch.device):
+    def to(self, device: torch.device, non_blocking: bool = False):
         # Boxes are assumed float32 and does not support to(dtype)
-        return RotatedBoxes(self.tensor.to(device=device))
+        return RotatedBoxes(self.tensor.to(device=device, non_blocking=non_blocking))
 
     def area(self) -> torch.Tensor:
         """