Skip to content

GitLab

Commit abeecb35 authored by Lara Haidar's avatar Lara Haidar Committed by Francisco Massa
Browse files

Disable Profiling in Failing Test (#1585) 

* disable test

* disable profiling

* Update test_onnx.py
parent 4b2f8dab
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
test/test_onnx.py
View file @ abeecb35
...@@ -299,6 +299,10 @@ class ONNXExporterTester(unittest.TestCase): ...@@ -299,6 +299,10 @@ class ONNXExporterTester(unittest.TestCase):
# 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)
......
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