diff --git a/src/nncf/onnx/graph/metatypes/groups.py b/src/nncf/onnx/graph/metatypes/groups.py index 32d490b2ed2..d772624dd06 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 dee4326542d..8874ed20a57 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 7fe8815edd4..995267f68a2 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 44242158c47..7602099cea1 100644 --- a/tests/post_training/requirements.txt +++ b/tests/post_training/requirements.txt @@ -7,6 +7,7 @@ torch torchvision onnx onnxruntime +onnxscript openvino pytest pytest-forked @@ -25,3 +26,4 @@ timm==0.9.2 accelerate==1.9.0 transformers==4.53.0 whowhatbench @ git+https://github.com/openvinotoolkit/openvino.genai@2025.4.1.0#subdirectory=tools/who_what_benchmark +datasets