Skip to content

Commit 691ec6d

Browse files
authored
Temporarily disable broken RCNN Onnx tests. (#3310)
1 parent d8eb22d commit 691ec6d

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

test/test_onnx.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -368,6 +368,7 @@ def get_test_images(self):
368368
test_images = [image2]
369369
return images, test_images
370370

371+
@unittest.skip # Skip until pytorch/vision#3251 is fixed and pytorch/pytorch#50910 is merged.
371372
def test_faster_rcnn(self):
372373
images, test_images = self.get_test_images()
373374
dummy_image = [torch.ones(3, 100, 100) * 0.3]
@@ -419,6 +420,7 @@ def test_paste_mask_in_image(self):
419420

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

423+
@unittest.skip # Skip until pytorch/vision#3251 is fixed and pytorch/pytorch#50910 is merged.
422424
def test_mask_rcnn(self):
423425
images, test_images = self.get_test_images()
424426
dummy_image = [torch.ones(3, 100, 100) * 0.3]
@@ -469,6 +471,7 @@ def test_heatmaps_to_keypoints(self):
469471
assert torch.all(out2[0].eq(out_trace2[0]))
470472
assert torch.all(out2[1].eq(out_trace2[1]))
471473

474+
@unittest.skip # Skip until pytorch/vision#3251 is fixed and pytorch/pytorch#50910 is merged.
472475
def test_keypoint_rcnn(self):
473476
images, test_images = self.get_test_images()
474477
dummy_images = [torch.ones(3, 100, 100) * 0.3]

0 commit comments

Comments
 (0)