Skip to content

Commit

Permalink
Update onnx, onnxruntime versions (#2601)
Browse files Browse the repository at this point in the history
### Changes

Pump up the version of ONNX and ONNXRuntime packages.

### Reason for changes

Many

### Related tickets

134294

### Tests

Updated
  • Loading branch information
kshpv authored Mar 26, 2024
1 parent cae3f02 commit 153d7e0
Show file tree
Hide file tree
Showing 21 changed files with 17,633 additions and 16,103 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,6 @@ torchvision
tqdm
scikit-learn
fastdownload
onnx~=1.13.1
onnxruntime~=1.14.1
onnx~=1.16.0
onnxruntime~=1.17.1
openvino-dev==2024.0
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
ultralytics==8.0.170
onnx>=1.12.0
onnx~=1.16.0
openvino-dev==2024.0
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -131,7 +131,7 @@ def find_version(*file_paths):
"torch==2.2.1;python_version < '3.11'",
]

ONNX_EXTRAS = ["onnx~=1.13.1", "onnxruntime~=1.14.1;python_version < '3.11'"]
ONNX_EXTRAS = ["onnx~=1.16.0", "onnxruntime~=1.17.1;python_version < '3.11'"]

OPENVINO_EXTRAS = ["openvino==2024.0"]

Expand Down
8,636 changes: 4,318 additions & 4,318 deletions tests/onnx/data/reference_graphs/original_nncf_graph/MaskRCNN-12.dot

Large diffs are not rendered by default.

298 changes: 149 additions & 149 deletions tests/onnx/data/reference_graphs/original_nncf_graph/fcn-resnet50-12.dot

Large diffs are not rendered by default.

Large diffs are not rendered by default.

1,771 changes: 943 additions & 828 deletions tests/onnx/data/reference_graphs/original_nncf_graph/ssd-12.dot

Large diffs are not rendered by default.

Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ strict digraph {
"3 nncf_model_input_1" [id=3, type=nncf_model_input];
"4 nncf_model_output_0" [id=4, type=nncf_model_output];
"5 nncf_model_output_1" [id=5, type=nncf_model_output];
"0 Add2" -> "4 nncf_model_output_0" [label="[2, 6, 3, 3", style=solid];
"0 Add2" -> "4 nncf_model_output_0" [label="[2, 6, 3, 3]", style=solid];
"1 Add1" -> "5 nncf_model_output_1" [label="[2, 6, 3, 3]", style=solid];
"2 nncf_model_input_0" -> "0 Add2" [label="[1, 6, 3, 3]", style=solid];
"2 nncf_model_input_0" -> "1 Add1" [label="[1, 6, 3, 3]", style=solid];
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,6 @@ strict digraph {
"0 Identity" -> "1 Embedding" [label="[10, 20]", style=solid];
"1 Embedding" -> "2 Gather" [label="[1, 10, 20]", style=solid];
"2 Gather" -> "3 MatMul" [label="[10, 20]", style=solid];
"3 MatMul" -> "5 nncf_model_output_0" [label="[1, 10]", style=solid];
"3 MatMul" -> "5 nncf_model_output_0" [label="[10, 10]", style=solid];
"4 nncf_model_input_0" -> "1 Embedding" [label="[1, 10]", style=dashed];
}
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ strict digraph {
"4 Cast1" -> "5 Cast2" [label="[]", style=dashed];
"5 Cast2" -> "6 Sqrt" [label="[]", style=solid];
"6 Sqrt" -> "7 Reshape" [label="[]", style=solid];
"7 Reshape" -> "8 Conv2" [label="[1]", style=solid];
"7 Reshape" -> "8 Conv2" [label="[]", style=solid];
"8 Conv2" -> "10 nncf_model_output_0" [label="[1, 10, 1, 1]", style=solid];
"9 nncf_model_input_0" -> "0 Conv1" [label="[1, 3, 32, 32]", style=solid];
}
7,030 changes: 3,755 additions & 3,275 deletions tests/onnx/data/reference_graphs/quantization/bertsquad-12.dot

Large diffs are not rendered by default.

Loading

0 comments on commit 153d7e0

Please sign in to comment.