Skip to content

Commit bc4d15f

Browse files
update opset version
1 parent 190a0f3 commit bc4d15f

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

tests/post_training/pipelines/image_classification_timm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ def prepare_model(self) -> None:
5252
self.dummy_tensor,
5353
onnx_path,
5454
export_params=True,
55-
opset_version=13,
55+
opset_version=18,
5656
**additional_kwargs,
5757
)
5858
self.model = onnx.load(onnx_path)

tests/post_training/pipelines/image_classification_torchvision.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -99,7 +99,7 @@ def prepare_model(self) -> None:
9999
self.dummy_tensor,
100100
onnx_path,
101101
export_params=True,
102-
opset_version=13,
102+
opset_version=18,
103103
**additional_kwargs,
104104
)
105105
self.model = onnx.load(onnx_path)

0 commit comments

Comments
 (0)