"git@developer.sourcefind.cn:one/TransferBench.git" did not exist on "9e3a04c3327972b041e3f9aefb7811b3f9b172b8"
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
......@@ -299,6 +299,10 @@ class ONNXExporterTester(unittest.TestCase):
# 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)
......
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