Switch off SQ for swin model in Torch backend

daniil-lyakhov · Jan 26, 2024 · 7e945b1 · 7e945b1
1 parent fc653c1
commit 7e945b1
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 6 deletions.
diff --git a/tests/post_training/model_scope.py b/tests/post_training/model_scope.py
@@ -218,7 +218,20 @@
             "preset": QuantizationPreset.MIXED,
             "model_type": ModelType.TRANSFORMER,
         },
-        "backends": ALL_PTQ_BACKENDS,
+        "backends": [BackendType.OV],
+    },
+    {
+        "reported_name": "timm/swin_base_patch4_window7_224_no_sq",
+        "model_id": "swin_base_patch4_window7_224",
+        "pipeline_cls": ImageClassificationTimm,
+        "ptq_params": {
+            "preset": QuantizationPreset.MIXED,
+            "model_type": ModelType.TRANSFORMER,
+            "advanced_parameters": AdvancedQuantizationParameters(
+                smooth_quant_alpha=AdvancedSmoothQuantParameters(matmul=-1)
+            ),
+        },
+        "backends": [BackendType.TORCH, BackendType.CUDA_TORCH, BackendType.ONNX],
     },
     {
         "reported_name": "timm/tf_inception_v3",

diff --git a/tests/post_training/reference_data.yaml b/tests/post_training/reference_data.yaml
@@ -176,15 +176,17 @@ timm/resnet18_backend_OV:
   metric_value: 0.71042
 timm/resnet18_backend_TORCH:
   metric_value: 0.71024
-timm/swin_base_patch4_window7_224_backend_CUDA_TORCH:
-  metric_value: 0.8131
 timm/swin_base_patch4_window7_224_backend_FP32:
   metric_value: 0.85274
-timm/swin_base_patch4_window7_224_backend_ONNX:
-  metric_value: 0.85158
 timm/swin_base_patch4_window7_224_backend_OV:
   metric_value: 0.83636
-timm/swin_base_patch4_window7_224_backend_TORCH:
+timm/swin_base_patch4_window7_224_no_sq_backend_FP32:
+  metric_value: 0.85274
+timm/swin_base_patch4_window7_224_no_sq_backend_CUDA_TORCH:
+  metric_value: 0.85142
+timm/swin_base_patch4_window7_224_no_sq_backend_ONNX:
+  metric_value: 0.85158
+timm/swin_base_patch4_window7_224_no_sq_backend_TORCH:
   metric_value: 0.85142
 timm/tf_inception_v3_backend_CUDA_TORCH:
   metric_value: 0.77542