Add advanced parameters to optimization config in DFine model

openvinotoolkit · Jan 3, 2025 · 7e010de · 7e010de
1 parent e3f7db2
commit 7e010de
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/otx/algo/detection/d_fine.py b/src/otx/algo/detection/d_fine.py
@@ -309,4 +309,12 @@ def _exporter(self) -> OTXModelExporter:
     @property
     def _optimization_config(self) -> dict[str, Any]:
         """PTQ config for D-FINE."""
-        return {"model_type": "transformer"}
+        return {
+            "model_type": "transformer",
+            "advanced_parameters": {
+                "activations_range_estimator_params": {
+                    "min": {"statistics_type": "QUANTILE", "aggregator_type": "MIN", "quantile_outlier_prob": 1e-4},
+                    "max": {"statistics_type": "QUANTILE", "aggregator_type": "MAX", "quantile_outlier_prob": 1e-4},
+                },
+            },
+        }