Skip to content

GitLab

Unverified Commit a3d87195 authored by Prabhat Roy's avatar Prabhat Roy Committed by GitHub
Browse files

Skip test_roi_align_aligned (#5029)

parent dcf5dc87
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
test/test_onnx.py
View file @ a3d87195
...@@ -141,6 +141,7 @@ class TestONNXExporter: ...@@ -141,6 +141,7 @@ class TestONNXExporter:
model = ops.RoIAlign((5, 5), 1, -1) model = ops.RoIAlign((5, 5), 1, -1)
self.run_model(model, [(x, single_roi)]) self.run_model(model, [(x, single_roi)])
@pytest.mark.skip(reason="ROIAlign with aligned=True is not supported in ONNX, but will be supported in opset 16.")
def test_roi_align_aligned(self): def test_roi_align_aligned(self):
x = torch.rand(1, 1, 10, 10, dtype=torch.float32) x = torch.rand(1, 1, 10, 10, dtype=torch.float32)
single_roi = torch.tensor([[0, 1.5, 1.5, 3, 3]], dtype=torch.float32) single_roi = torch.tensor([[0, 1.5, 1.5, 3, 3]], dtype=torch.float32)
......
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