diff --git a/test/test_ops.py b/test/test_ops.py index 743fe159e37..aad4bff2085 100644 --- a/test/test_ops.py +++ b/test/test_ops.py @@ -95,6 +95,7 @@ def forward(self, imgs: Tensor, boxes: List[Tensor]) -> Tensor: class RoIOpTester(ABC): + # Blop dtype = torch.float64 mps_dtype = torch.float32 mps_backward_atol = 2e-2