Enhance training schedule for multi-label classification

external/model-preparation-algorithm/configs/classification/efficientnet_b0_cls_incr/model_multilabel.py

@@ -1,4 +1,18 @@
 _base_ = [
   '../../../submodule/models/classification/ote_efficientnet_b0_multilabel.yaml',
 ]
+
+runner = dict(max_epochs=90)
 fp16 = dict(loss_scale=512.)
+
+custom_hooks = [
+    dict(
+        type='LazyEarlyStoppingHook',
+        start=0,
+        patience=10,
+        iteration_patience=0,
+        metric='mAP',
+        interval=1,
+        priority=75,
+    ),
+]

external/model-preparation-algorithm/configs/classification/efficientnet_b0_cls_incr/template.yaml

@@ -34,6 +34,8 @@ hyper_parameters:
       learning_rate:
         default_value: 0.007
         auto_hpo_state: POSSIBLE
+      learning_rate_warmup_iters:
+        default_value: 0
       num_iters:
         default_value: 20
     algo_backend:

external/model-preparation-algorithm/configs/classification/efficientnet_v2_s_cls_incr/model_multilabel.py

@@ -1,4 +1,18 @@
 _base_ = [
   '../../../submodule/models/classification/ote_efficientnet_v2_s_multilabel.yaml',
 ]
+
+runner = dict(max_epochs=90)
 fp16 = dict(loss_scale=512.)
+
+custom_hooks = [
+    dict(
+        type='LazyEarlyStoppingHook',
+        start=0,
+        patience=10,
+        iteration_patience=0,
+        metric='mAP',
+        interval=1,
+        priority=75,
+    ),
+]

external/model-preparation-algorithm/configs/classification/efficientnet_v2_s_cls_incr/template.yaml

@@ -34,6 +34,8 @@ hyper_parameters:
       learning_rate:
         default_value: 0.007
         auto_hpo_state: POSSIBLE
+      learning_rate_warmup_iters:
+        default_value: 0
       num_iters:
         default_value: 20
     algo_backend:

external/model-preparation-algorithm/configs/classification/mobilenet_v3_large_1_cls_incr/model_multilabel.py

@@ -1,4 +1,18 @@
 _base_ = [
   '../../../submodule/models/classification/ote_mobilenet_v3_large_multilabel.yaml',
 ]
+
+runner = dict(max_epochs=90)
 fp16 = dict(loss_scale=512.)
+
+custom_hooks = [
+    dict(
+        type='LazyEarlyStoppingHook',
+        start=0,
+        patience=10,
+        iteration_patience=0,
+        metric='mAP',
+        interval=1,
+        priority=75,
+    ),
+]

external/model-preparation-algorithm/mpa_tasks/apis/classification/task.py

@@ -300,6 +300,14 @@ def _init_model_cfg(self):
         else:
             cfg_path = os.path.join(base_dir, 'model.py')
         cfg = MPAConfig.fromfile(cfg_path)
+
+        # To initialize different HP for multi-label classification / Support HP change via CLI & UI
+        if self._multilabel:
+            template = MPAConfig.fromfile(self.template_file_path)
+            template_params = template.hyper_parameters.parameter_overrides.learning_parameters
+            if template_params.num_iters.default_value != self._hyperparams.learning_parameters.num_iters:
+                cfg.pop('runner', False)
+
         cfg.model.multilabel = self._multilabel
         cfg.model.hierarchical = self._hierarchical
         if self._hierarchical: