Unverified Commit a3d9c416 authored by Vasilis Vryniotis's avatar Vasilis Vryniotis Committed by GitHub
Browse files

Do not disable profiling executor in ONNX tests (#4324)



[ghstack-poisoned]
Co-authored-by: default avatarBert Maher <bertrand@fb.com>
parent 96f6e0a1
...@@ -397,10 +397,6 @@ class TestONNXExporter: ...@@ -397,10 +397,6 @@ class TestONNXExporter:
# This test also compares both paste_masks_in_image and _onnx_paste_masks_in_image # 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). # (since jit_trace witll call _onnx_paste_masks_in_image).
def test_paste_mask_in_image(self): 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) masks = torch.rand(10, 1, 26, 26)
boxes = torch.rand(10, 4) boxes = torch.rand(10, 4)
boxes[:, 2:] += torch.rand(10, 2) boxes[:, 2:] += torch.rand(10, 2)
...@@ -452,10 +448,6 @@ class TestONNXExporter: ...@@ -452,10 +448,6 @@ class TestONNXExporter:
# This test also compares both heatmaps_to_keypoints and _onnx_heatmaps_to_keypoints # This test also compares both heatmaps_to_keypoints and _onnx_heatmaps_to_keypoints
# (since jit_trace witll call _heatmaps_to_keypoints). # (since jit_trace witll call _heatmaps_to_keypoints).
def test_heatmaps_to_keypoints(self): def test_heatmaps_to_keypoints(self):
# disable profiling
torch._C._jit_set_profiling_executor(False)
torch._C._jit_set_profiling_mode(False)
maps = torch.rand(10, 1, 26, 26) maps = torch.rand(10, 1, 26, 26)
rois = torch.rand(10, 4) rois = torch.rand(10, 4)
from torchvision.models.detection.roi_heads import heatmaps_to_keypoints from torchvision.models.detection.roi_heads import heatmaps_to_keypoints
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment