Skip to content

Commit

Permalink
skip test_imagenet
Browse files Browse the repository at this point in the history
  • Loading branch information
mshr-h committed Sep 23, 2024
1 parent 0606261 commit dfab291
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions tests/python/frontend/pytorch/test_fx_quant.py
Original file line number Diff line number Diff line change
Expand Up @@ -87,6 +87,9 @@ def forward(self, inp):
quantize_and_build(model, 300)


@pytest.mark.skip(
reason="Model binary isn't uploaded to S3. See https://github.com/apache/tvm/pull/17397"
)
def test_imagenet():
for model_func in [resnet50, efficientnet_b4]:
quantize_and_build(model_func(pretrained=True).eval(), 224)

0 comments on commit dfab291

Please sign in to comment.