This is an automated email from the ASF dual-hosted git repository. leandron pushed a commit to branch ci-docker-staging in repository https://gitbox.apache.org/repos/asf/tvm.git
commit 4b66fb99f6fd54816c8a50525745b16f2b20933c Author: Leandro Nunes <[email protected]> AuthorDate: Mon Aug 22 11:52:27 2022 +0100 Apply ONNX fix Change-Id: I1f5dd944e72964cf91245dec452250d6d4d0739a --- tests/python/driver/tvmc/test_frontends.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/python/driver/tvmc/test_frontends.py b/tests/python/driver/tvmc/test_frontends.py index 1e6efb4a3b..0cd02181ac 100644 --- a/tests/python/driver/tvmc/test_frontends.py +++ b/tests/python/driver/tvmc/test_frontends.py @@ -169,7 +169,7 @@ def verify_load_model__onnx(model, **kwargs): def test_load_model__onnx(onnx_resnet50): # some CI environments wont offer onnx, so skip in case it is not present pytest.importorskip("onnx") - tvmc_model = verify_load_model__onnx(onnx_resnet50) + tvmc_model = verify_load_model__onnx(onnx_resnet50, freeze_params=False) # check whether one known value is part of the params dict assert "resnetv24_batchnorm0_gamma" in tvmc_model.params.keys() tvmc_model = verify_load_model__onnx(onnx_resnet50, freeze_params=True)
