Skip to content

GitLab

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

Temporarily disable broken RCNN Onnx tests. (#3310)

parent d8eb22db
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
test/test_onnx.py
View file @ 691ec6dd
...@@ -368,6 +368,7 @@ class ONNXExporterTester(unittest.TestCase): ...@@ -368,6 +368,7 @@ class ONNXExporterTester(unittest.TestCase):
test_images = [image2] test_images = [image2]
return images, test_images return images, test_images
@unittest.skip # Skip until pytorch/vision#3251 is fixed and pytorch/pytorch#50910 is merged.
def test_faster_rcnn(self): def test_faster_rcnn(self):
images, test_images = self.get_test_images() images, test_images = self.get_test_images()
dummy_image = [torch.ones(3, 100, 100) * 0.3] dummy_image = [torch.ones(3, 100, 100) * 0.3]
...@@ -419,6 +420,7 @@ class ONNXExporterTester(unittest.TestCase): ...@@ -419,6 +420,7 @@ class ONNXExporterTester(unittest.TestCase):
assert torch.all(out2.eq(out_trace2)) assert torch.all(out2.eq(out_trace2))
@unittest.skip # Skip until pytorch/vision#3251 is fixed and pytorch/pytorch#50910 is merged.
def test_mask_rcnn(self): def test_mask_rcnn(self):
images, test_images = self.get_test_images() images, test_images = self.get_test_images()
dummy_image = [torch.ones(3, 100, 100) * 0.3] dummy_image = [torch.ones(3, 100, 100) * 0.3]
...@@ -469,6 +471,7 @@ class ONNXExporterTester(unittest.TestCase): ...@@ -469,6 +471,7 @@ class ONNXExporterTester(unittest.TestCase):
assert torch.all(out2[0].eq(out_trace2[0])) assert torch.all(out2[0].eq(out_trace2[0]))
assert torch.all(out2[1].eq(out_trace2[1])) assert torch.all(out2[1].eq(out_trace2[1]))
@unittest.skip # Skip until pytorch/vision#3251 is fixed and pytorch/pytorch#50910 is merged.
def test_keypoint_rcnn(self): def test_keypoint_rcnn(self):
images, test_images = self.get_test_images() images, test_images = self.get_test_images()
dummy_images = [torch.ones(3, 100, 100) * 0.3] dummy_images = [torch.ones(3, 100, 100) * 0.3]
......
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