Skip to content

Commit

Permalink
Update URL for real models from ONNX Runtime (onnx#4865)
Browse files Browse the repository at this point in the history
* Update URL for real models from ONNX Runtime

Signed-off-by: Chun-Wei Chen <jacky82226@gmail.com>

* update missing BASE_URL

Signed-off-by: Chun-Wei Chen <jacky82226@gmail.com>

* fix black

Signed-off-by: Chun-Wei Chen <jacky82226@gmail.com>

* last / :(

Signed-off-by: Chun-Wei Chen <jacky82226@gmail.com>

---------

Signed-off-by: Chun-Wei Chen <jacky82226@gmail.com>
  • Loading branch information
jcwchen authored and Bjarke Roune committed May 6, 2023
1 parent 4498209 commit a992a68
Show file tree
Hide file tree
Showing 10 changed files with 10 additions and 13 deletions.
5 changes: 1 addition & 4 deletions onnx/backend/test/case/model/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -35,10 +35,7 @@ def expect(
)


base_model_opset_version = 10
BASE_URL = (
f"https://s3.amazonaws.com/download.onnx/models/opset_{base_model_opset_version}"
)
BASE_URL = "https://download.onnxruntime.ai/onnx/models"


def collect_testcases() -> List[TestCase]:
Expand Down
2 changes: 1 addition & 1 deletion onnx/backend/test/data/real/test_bvlc_alexnet/data.json
Original file line number Diff line number Diff line change
@@ -1 +1 @@
{"atol": 1e-07, "model_name": "bvlc_alexnet", "rtol": 0.001, "url": "https://s3.amazonaws.com/download.onnx/models/opset_10/bvlc_alexnet.tar.gz"}
{"atol": 1e-07, "model_name": "bvlc_alexnet", "rtol": 0.001, "url": "https://download.onnxruntime.ai/onnx/models/bvlc_alexnet.tar.gz"}
2 changes: 1 addition & 1 deletion onnx/backend/test/data/real/test_densenet121/data.json
Original file line number Diff line number Diff line change
@@ -1 +1 @@
{"atol": 1e-07, "model_name": "densenet121", "rtol": 0.002, "url": "https://s3.amazonaws.com/download.onnx/models/opset_10/densenet121.tar.gz"}
{"atol": 1e-07, "model_name": "densenet121", "rtol": 0.002, "url": "https://download.onnxruntime.ai/onnx/models/densenet121.tar.gz"}
2 changes: 1 addition & 1 deletion onnx/backend/test/data/real/test_inception_v1/data.json
Original file line number Diff line number Diff line change
@@ -1 +1 @@
{"atol": 1e-07, "model_name": "inception_v1", "rtol": 0.001, "url": "https://s3.amazonaws.com/download.onnx/models/opset_10/inception_v1.tar.gz"}
{"atol": 1e-07, "model_name": "inception_v1", "rtol": 0.001, "url": "https://download.onnxruntime.ai/onnx/models/inception_v1.tar.gz"}
2 changes: 1 addition & 1 deletion onnx/backend/test/data/real/test_inception_v2/data.json
Original file line number Diff line number Diff line change
@@ -1 +1 @@
{"atol": 1e-07, "model_name": "inception_v2", "rtol": 0.001, "url": "https://s3.amazonaws.com/download.onnx/models/opset_10/inception_v2.tar.gz"}
{"atol": 1e-07, "model_name": "inception_v2", "rtol": 0.001, "url": "https://download.onnxruntime.ai/onnx/models/inception_v2.tar.gz"}
2 changes: 1 addition & 1 deletion onnx/backend/test/data/real/test_resnet50/data.json
Original file line number Diff line number Diff line change
@@ -1 +1 @@
{"atol": 1e-07, "model_name": "resnet50", "rtol": 0.001, "url": "https://s3.amazonaws.com/download.onnx/models/opset_10/resnet50.tar.gz"}
{"atol": 1e-07, "model_name": "resnet50", "rtol": 0.001, "url": "https://download.onnxruntime.ai/onnx/models/resnet50.tar.gz"}
2 changes: 1 addition & 1 deletion onnx/backend/test/data/real/test_shufflenet/data.json
Original file line number Diff line number Diff line change
@@ -1 +1 @@
{"atol": 1e-07, "model_name": "shufflenet", "rtol": 0.001, "url": "https://s3.amazonaws.com/download.onnx/models/opset_10/shufflenet.tar.gz"}
{"atol": 1e-07, "model_name": "shufflenet", "rtol": 0.001, "url": "https://download.onnxruntime.ai/onnx/models/shufflenet.tar.gz"}
2 changes: 1 addition & 1 deletion onnx/backend/test/data/real/test_squeezenet/data.json
Original file line number Diff line number Diff line change
@@ -1 +1 @@
{"atol": 1e-07, "model_name": "squeezenet", "rtol": 0.001, "url": "https://s3.amazonaws.com/download.onnx/models/opset_10/squeezenet.tar.gz"}
{"atol": 1e-07, "model_name": "squeezenet", "rtol": 0.001, "url": "https://download.onnxruntime.ai/onnx/models/squeezenet.tar.gz"}
2 changes: 1 addition & 1 deletion onnx/backend/test/data/real/test_vgg19/data.json
Original file line number Diff line number Diff line change
@@ -1 +1 @@
{"atol": 1e-07, "model_name": "vgg19", "rtol": 0.001, "url": "https://s3.amazonaws.com/download.onnx/models/opset_10/vgg19.tar.gz"}
{"atol": 1e-07, "model_name": "vgg19", "rtol": 0.001, "url": "https://download.onnxruntime.ai/onnx/models/vgg19.tar.gz"}
2 changes: 1 addition & 1 deletion onnx/backend/test/data/real/test_zfnet512/data.json
Original file line number Diff line number Diff line change
@@ -1 +1 @@
{"atol": 1e-07, "model_name": "zfnet512", "rtol": 0.001, "url": "https://s3.amazonaws.com/download.onnx/models/opset_10/zfnet512.tar.gz"}
{"atol": 1e-07, "model_name": "zfnet512", "rtol": 0.001, "url": "https://download.onnxruntime.ai/onnx/models/zfnet512.tar.gz"}

0 comments on commit a992a68

Please sign in to comment.