fix precision for onnly pruning

openvinotoolkit · Jun 2, 2022 · 561daf0 · 561daf0
1 parent e9e58f9
commit 561daf0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/external/mmdetection/detection_tasks/apis/detection/nncf_task.py b/external/mmdetection/detection_tasks/apis/detection/nncf_task.py
@@ -91,7 +91,7 @@ def _set_attributes_by_hyperparams(self):
         if not quantization and pruning:
             self._nncf_preset = "nncf_pruning"
             self._optimization_methods = [OptimizationMethod.FILTER_PRUNING]
-            self._precision = [ModelPrecision.INT8]
+            self._precision = self._precision_from_config
             return
         raise RuntimeError('Not selected optimization algorithm')