Fix QAT for detection tiling models

openvinotoolkit · Apr 23, 2024 · d487449 · d487449
1 parent aa611ab
commit d487449
Show file tree

Hide file tree

Showing 6 changed files with 52 additions and 75 deletions.
diff --git a/src/otx/algorithms/detection/configs/detection/cspdarknet_yolox_l/compression_config.json b/src/otx/algorithms/detection/configs/detection/cspdarknet_yolox_l/compression_config.json
@@ -17,7 +17,16 @@
     "nncf_config": {
       "compression": [
         {
-          "algorithm": "quantization"
+          "algorithm": "quantization",
+          "initializer": {
+            "range": {
+              "type": "percentile",
+              "params": {
+                "min_percentile": 0,
+                "max_percentile": 100
+              }
+            }
+          }
         }
       ],
       "accuracy_aware_training": {

diff --git a/src/otx/algorithms/detection/configs/detection/cspdarknet_yolox_s/compression_config.json b/src/otx/algorithms/detection/configs/detection/cspdarknet_yolox_s/compression_config.json
@@ -17,7 +17,16 @@
     "nncf_config": {
       "compression": [
         {
-          "algorithm": "quantization"
+          "algorithm": "quantization",
+          "initializer": {
+            "range": {
+              "type": "percentile",
+              "params": {
+                "min_percentile": 0,
+                "max_percentile": 100
+              }
+            }
+          }
         }
       ],
       "accuracy_aware_training": {

diff --git a/src/otx/algorithms/detection/configs/detection/cspdarknet_yolox_tiny/compression_config.json b/src/otx/algorithms/detection/configs/detection/cspdarknet_yolox_tiny/compression_config.json
@@ -17,7 +17,16 @@
     "nncf_config": {
       "compression": [
         {
-          "algorithm": "quantization"
+          "algorithm": "quantization",
+          "initializer": {
+            "range": {
+              "type": "percentile",
+              "params": {
+                "min_percentile": 0,
+                "max_percentile": 100
+              }
+            }
+          }
         }
       ],
       "accuracy_aware_training": {

diff --git a/src/otx/algorithms/detection/configs/detection/cspdarknet_yolox_x/compression_config.json b/src/otx/algorithms/detection/configs/detection/cspdarknet_yolox_x/compression_config.json
@@ -17,7 +17,16 @@
     "nncf_config": {
       "compression": [
         {
-          "algorithm": "quantization"
+          "algorithm": "quantization",
+          "initializer": {
+            "range": {
+              "type": "percentile",
+              "params": {
+                "min_percentile": 0,
+                "max_percentile": 100
+              }
+            }
+          }
         }
       ],
       "accuracy_aware_training": {

diff --git a/src/otx/algorithms/detection/configs/detection/mobilenetv2_atss/compression_config.json b/src/otx/algorithms/detection/configs/detection/mobilenetv2_atss/compression_config.json
@@ -20,6 +20,11 @@
           "algorithm": "quantization",
           "initializer": {
             "range": {
+              "type": "percentile",
+              "params": {
+                "min_percentile": 0,
+                "max_percentile": 100
+              },
               "num_init_samples": 300
             },
             "batchnorm_adaptation": {
@@ -37,38 +42,5 @@
       }
     }
   },
-  "nncf_quantization_pruning": {
-    "nncf_config": {
-      "accuracy_aware_training": {
-        "mode": "adaptive_compression_level",
-        "params": {
-          "initial_training_phase_epochs": 5,
-          "maximal_total_epochs": 100,
-          "patience_epochs": 5
-        }
-      },
-      "compression": [
-        {
-          "algorithm": "filter_pruning",
-          "params": {
-            "schedule": "baseline",
-            "pruning_flops_target": 0.1,
-            "filter_importance": "geometric_median"
-          }
-        },
-        {
-          "algorithm": "quantization",
-          "initializer": {
-            "range": {
-              "num_init_samples": 300
-            },
-            "batchnorm_adaptation": {
-              "num_bn_adaptation_samples": 300
-            }
-          }
-        }
-      ]
-    }
-  },
-  "order_of_parts": ["nncf_quantization", "nncf_quantization_pruning"]
+  "order_of_parts": ["nncf_quantization"]
 }
diff --git a/src/otx/algorithms/detection/configs/detection/mobilenetv2_ssd/compression_config.json b/src/otx/algorithms/detection/configs/detection/mobilenetv2_ssd/compression_config.json
@@ -20,6 +20,11 @@
           "algorithm": "quantization",
           "initializer": {
             "range": {
+              "type": "percentile",
+              "params": {
+                "min_percentile": 0,
+                "max_percentile": 100
+              },
               "num_init_samples": 300
             },
             "batchnorm_adaptation": {
@@ -37,41 +42,5 @@
       }
     }
   },
-  "nncf_quantization_pruning": {
-    "nncf_config": {
-      "accuracy_aware_training": {
-        "mode": "adaptive_compression_level",
-        "params": {
-          "initial_training_phase_epochs": 5,
-          "maximal_total_epochs": 100,
-          "patience_epochs": 5
-        }
-      },
-      "compression": [
-        {
-          "algorithm": "filter_pruning",
-          "ignored_scopes": [
-            "{re}CustomSingleStageDetector/CustomSSDHead\\[bbox_head\\].*"
-          ],
-          "params": {
-            "schedule": "baseline",
-            "pruning_flops_target": 0.1,
-            "filter_importance": "geometric_median"
-          }
-        },
-        {
-          "algorithm": "quantization",
-          "initializer": {
-            "range": {
-              "num_init_samples": 300
-            },
-            "batchnorm_adaptation": {
-              "num_bn_adaptation_samples": 300
-            }
-          }
-        }
-      ]
-    }
-  },
-  "order_of_parts": ["nncf_quantization", "nncf_quantization_pruning"]
+  "order_of_parts": ["nncf_quantization"]
 }