diff --git a/test/test_transforms_v2.py b/test/test_transforms_v2.py index 62077be1db5..487018b7c69 100644 --- a/test/test_transforms_v2.py +++ b/test/test_transforms_v2.py @@ -1439,6 +1439,7 @@ def test_kernel_bounding_boxes(self, param, value, format, dtype, device): canvas_size=bounding_boxes.canvas_size, **{param: value}, check_scripted_vs_eager=not (param == "shear" and isinstance(value, (int, float))), + check_cuda_vs_cpu=dict(atol=1e-5, rtol=1e-5), ) @param_value_parametrization(