diff --git a/test/test_onnx.py b/test/test_onnx.py index 090f16cc550..a42d88103df 100644 --- a/test/test_onnx.py +++ b/test/test_onnx.py @@ -299,6 +299,10 @@ def test_faster_rcnn(self): # This test also compares both paste_masks_in_image and _onnx_paste_masks_in_image # (since jit_trace witll call _onnx_paste_masks_in_image). def test_paste_mask_in_image(self): + # disable profiling + torch._C._jit_set_profiling_executor(False) + torch._C._jit_set_profiling_mode(False) + masks = torch.rand(10, 1, 26, 26) boxes = torch.rand(10, 4) boxes[:, 2:] += torch.rand(10, 2)