Change default from multi-class to multi-label

openvinotoolkit · Aug 25, 2022 · a8085bc · a8085bc
1 parent 33a0874
commit a8085bc
Show file tree

Hide file tree

Showing 10 changed files with 11 additions and 8 deletions.
diff --git a/...rnal/model-preparation-algorithm/configs/classification/efficientnet_b0_cls_incr/model.py b/...rnal/model-preparation-algorithm/configs/classification/efficientnet_b0_cls_incr/model.py
@@ -1,4 +1,6 @@
 _base_ = [
   '../../../submodule/models/classification/ote_efficientnet_b0.yaml',
 ]
+
+runner = dict(max_epochs=20)
 fp16 = dict(loss_scale=512.)
diff --git a/...preparation-algorithm/configs/classification/efficientnet_b0_cls_incr/model_multilabel.py b/...preparation-algorithm/configs/classification/efficientnet_b0_cls_incr/model_multilabel.py
@@ -2,7 +2,6 @@
   '../../../submodule/models/classification/ote_efficientnet_b0_multilabel.yaml',
 ]
 
-runner = dict(max_epochs=90)
 fp16 = dict(loss_scale=512.)
 
 custom_hooks = [

diff --git a/...model-preparation-algorithm/configs/classification/efficientnet_b0_cls_incr/template.yaml b/...model-preparation-algorithm/configs/classification/efficientnet_b0_cls_incr/template.yaml
@@ -37,7 +37,7 @@ hyper_parameters:
       learning_rate_warmup_iters:
         default_value: 0
       num_iters:
-        default_value: 20
+        default_value: 90
     algo_backend:
       train_type:
         default_value: Incremental

diff --git a/...al/model-preparation-algorithm/configs/classification/efficientnet_v2_s_cls_incr/model.py b/...al/model-preparation-algorithm/configs/classification/efficientnet_v2_s_cls_incr/model.py
@@ -1,4 +1,6 @@
 _base_ = [
   '../../../submodule/models/classification/ote_efficientnet_v2_s.yaml',
 ]
+
+runner = dict(max_epochs=20)
 fp16 = dict(loss_scale=512.)
diff --git a/...eparation-algorithm/configs/classification/efficientnet_v2_s_cls_incr/model_multilabel.py b/...eparation-algorithm/configs/classification/efficientnet_v2_s_cls_incr/model_multilabel.py
@@ -2,7 +2,6 @@
   '../../../submodule/models/classification/ote_efficientnet_v2_s_multilabel.yaml',
 ]
 
-runner = dict(max_epochs=90)
 fp16 = dict(loss_scale=512.)
 
 custom_hooks = [

diff --git a/...del-preparation-algorithm/configs/classification/efficientnet_v2_s_cls_incr/template.yaml b/...del-preparation-algorithm/configs/classification/efficientnet_v2_s_cls_incr/template.yaml
@@ -37,7 +37,7 @@ hyper_parameters:
       learning_rate_warmup_iters:
         default_value: 0
       num_iters:
-        default_value: 20
+        default_value: 90
     algo_backend:
       train_type:
         default_value: Incremental

diff --git a/...model-preparation-algorithm/configs/classification/mobilenet_v3_large_1_cls_incr/model.py b/...model-preparation-algorithm/configs/classification/mobilenet_v3_large_1_cls_incr/model.py
@@ -1,4 +1,6 @@
 _base_ = [
   '../../../submodule/models/classification/ote_mobilenet_v3_large.yaml',
 ]
+
+runner = dict(max_epochs=20)
 fp16 = dict(loss_scale=512.)
diff --git a/...ration-algorithm/configs/classification/mobilenet_v3_large_1_cls_incr/model_multilabel.py b/...ration-algorithm/configs/classification/mobilenet_v3_large_1_cls_incr/model_multilabel.py
@@ -2,7 +2,6 @@
   '../../../submodule/models/classification/ote_mobilenet_v3_large_multilabel.yaml',
 ]
 
-runner = dict(max_epochs=90)
 fp16 = dict(loss_scale=512.)
 
 custom_hooks = [

diff --git a/...-preparation-algorithm/configs/classification/mobilenet_v3_large_1_cls_incr/template.yaml b/...-preparation-algorithm/configs/classification/mobilenet_v3_large_1_cls_incr/template.yaml
@@ -37,7 +37,7 @@ hyper_parameters:
       learning_rate_warmup_iters:
         default_value: 100
       num_iters:
-        default_value: 20
+        default_value: 90
     algo_backend:
       train_type:
         default_value: Incremental

diff --git a/external/model-preparation-algorithm/mpa_tasks/apis/classification/task.py b/external/model-preparation-algorithm/mpa_tasks/apis/classification/task.py
@@ -301,8 +301,8 @@ def _init_model_cfg(self):
             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:
+        # To initialize different HP according to task / Support HP change via CLI & UI
+        if not self._multilabel:
             template = MPAConfig.fromfile(self.template_file_path)
             template_params = template.hyper_parameters.parameter_overrides.learning_parameters
             incoming_params = self._hyperparams.learning_parameters