From 7e010de31251665d23492eb3f870756ad2c3e4f3 Mon Sep 17 00:00:00 2001 From: Eugene Liu Date: Fri, 3 Jan 2025 16:10:29 +0000 Subject: [PATCH] Add advanced parameters to optimization config in DFine model --- src/otx/algo/detection/d_fine.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/src/otx/algo/detection/d_fine.py b/src/otx/algo/detection/d_fine.py index adbdc3b843..ba282e08a0 100644 --- 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}, + }, + }, + }