Skip to content

Commit

Permalink
[ONNX] Disable model tests since export of interpolate script module …
Browse files Browse the repository at this point in the history
…is broken (#1989)

* Fix for roi_align export

* Disable interpolate script module tests

Disable test until export of interpolate script module to ONNX is fixed
  • Loading branch information
neginraoof authored Mar 17, 2020
1 parent e96f2d5 commit 8b9859d
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion test/test_onnx.py
Original file line number Diff line number Diff line change
Expand Up @@ -343,6 +343,7 @@ def test_paste_mask_in_image(self):

assert torch.all(out2.eq(out_trace2))

@unittest.skip("Disable test until export of interpolate script module to ONNX is fixed")
def test_mask_rcnn(self):
images, test_images = self.get_test_images()

Expand Down Expand Up @@ -378,7 +379,8 @@ def test_heatmaps_to_keypoints(self):

assert torch.all(out2[0].eq(out_trace2[0]))
assert torch.all(out2[1].eq(out_trace2[1]))


@unittest.skip("Disable test until export of interpolate script module to ONNX is fixed")
def test_keypoint_rcnn(self):
class KeyPointRCNN(torch.nn.Module):
def __init__(self):
Expand Down

0 comments on commit 8b9859d

Please sign in to comment.