diff --git a/test/test_transforms_v2.py b/test/test_transforms_v2.py index a87e601e1a6..2d63686f504 100644 --- a/test/test_transforms_v2.py +++ b/test/test_transforms_v2.py @@ -943,7 +943,7 @@ def test_bounding_boxes_correctness(self, format, size, use_max_size, fn): expected = self._reference_resize_bounding_boxes(bounding_boxes, format=format, size=size, **max_size_kwarg) self._check_output_size(bounding_boxes, actual, size=size, **max_size_kwarg) - torch.testing.assert_close(actual, expected) + torch.testing.assert_close(actual, expected, atol=1.1e-5, rtol=1e-3) def _reference_resize_keypoints(self, keypoints, *, size, max_size=None): old_height, old_width = keypoints.canvas_size