diff --git a/src/nncf/onnx/graph/metatypes/groups.py b/src/nncf/onnx/graph/metatypes/groups.py index 758b61eee08..514b5ac3040 100644 --- a/src/nncf/onnx/graph/metatypes/groups.py +++ b/src/nncf/onnx/graph/metatypes/groups.py @@ -54,6 +54,7 @@ onnx_metatypes.ONNXDepthwiseConvolutionMetatype, onnx_metatypes.ONNXConvolutionTransposeMetatype, onnx_metatypes.ONNXDeformableConvolutionMetatype, + onnx_metatypes.ONNXGroupConvolutionMetatype, ] LINEAR_OPERATIONS = [*CONVOLUTION_METATYPES, *MATMUL_METATYPES] diff --git a/tests/post_training/pipelines/image_classification_timm.py b/tests/post_training/pipelines/image_classification_timm.py index 8f2d53b8fb1..8ba17f7b49a 100644 --- a/tests/post_training/pipelines/image_classification_timm.py +++ b/tests/post_training/pipelines/image_classification_timm.py @@ -52,8 +52,7 @@ def prepare_model(self) -> None: self.dummy_tensor, onnx_path, export_params=True, - opset_version=13, - dynamo=False, + opset_version=18, **additional_kwargs, ) self.model = onnx.load(onnx_path) diff --git a/tests/post_training/pipelines/image_classification_torchvision.py b/tests/post_training/pipelines/image_classification_torchvision.py index 781dfc6f3ad..8a575c2dc1c 100644 --- a/tests/post_training/pipelines/image_classification_torchvision.py +++ b/tests/post_training/pipelines/image_classification_torchvision.py @@ -99,8 +99,7 @@ def prepare_model(self) -> None: self.dummy_tensor, onnx_path, export_params=True, - opset_version=13, - dynamo=False, + opset_version=18, **additional_kwargs, ) self.model = onnx.load(onnx_path) diff --git a/tests/post_training/requirements.txt b/tests/post_training/requirements.txt index 0a9609ed390..1227a4df7ff 100644 --- a/tests/post_training/requirements.txt +++ b/tests/post_training/requirements.txt @@ -3,6 +3,7 @@ torch torchvision onnx onnxruntime +onnxscript openvino pytest pytest-forked