Edit num_workers and change MPA repo as a latest (#1314)

openvinotoolkit · Oct 28, 2022 · 91a1be1 · 91a1be1
1 parent 89754c1
commit 91a1be1
Show file tree

Hide file tree

Showing 8 changed files with 3 additions and 13 deletions.
diff --git a/external/model-preparation-algorithm/configs/classification/configuration.yaml b/external/model-preparation-algorithm/configs/classification/configuration.yaml
@@ -83,7 +83,7 @@ learning_parameters:
     warning: null
   num_workers:
     affects_outcome_of: NONE
-    default_value: 0
+    default_value: 2
     description:
       Increasing this value might improve training speed however it might
       cause out of memory errors. If the number of workers is set to zero, data loading

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
@@ -29,8 +29,6 @@ hyper_parameters:
       batch_size:
         default_value: 64
         auto_hpo_state: POSSIBLE
-      num_workers:
-        default_value: 0
       learning_rate:
         default_value: 0.0049
         auto_hpo_state: POSSIBLE

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
@@ -29,8 +29,6 @@ hyper_parameters:
       batch_size:
         default_value: 64
         auto_hpo_state: POSSIBLE
-      num_workers:
-        default_value: 0
       learning_rate:
         default_value: 0.0071
         auto_hpo_state: POSSIBLE

diff --git a/...algorithm/configs/classification/mobilenet_v3_large_075_cls_incr/template_experiment.yaml b/...algorithm/configs/classification/mobilenet_v3_large_075_cls_incr/template_experiment.yaml
@@ -29,8 +29,6 @@ hyper_parameters:
       batch_size:
         default_value: 32
         auto_hpo_state: POSSIBLE
-      num_workers:
-        default_value: 4
       learning_rate:
         default_value: 0.016
         auto_hpo_state: POSSIBLE

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
@@ -29,8 +29,6 @@ hyper_parameters:
       batch_size:
         default_value: 64
         auto_hpo_state: POSSIBLE
-      num_workers:
-        default_value: 0
       learning_rate:
         default_value: 0.0058
         auto_hpo_state: POSSIBLE

diff --git a/...ion-algorithm/configs/classification/mobilenet_v3_small_cls_incr/template_experiment.yaml b/...ion-algorithm/configs/classification/mobilenet_v3_small_cls_incr/template_experiment.yaml
@@ -29,8 +29,6 @@ hyper_parameters:
       batch_size:
         default_value: 32
         auto_hpo_state: POSSIBLE
-      num_workers:
-        default_value: 4
       learning_rate:
         default_value: 0.016
         auto_hpo_state: POSSIBLE

diff --git a/external/model-preparation-algorithm/configs/detection/configuration.yaml b/external/model-preparation-algorithm/configs/detection/configuration.yaml
@@ -101,7 +101,7 @@ learning_parameters:
     warning: null
   num_workers:
     affects_outcome_of: NONE
-    default_value: 0
+    default_value: 2
     description:
       Increasing this value might improve training speed however it might
       cause out of memory errors. If the number of workers is set to zero, data loading

diff --git a/external/model-preparation-algorithm/submodule b/external/model-preparation-algorithm/submodule
+7 −0		mpa/det/trainer.py
+4 −0		recipes/stages/_base_/runners/cls_runner.py
+5 −4		recipes/stages/classification/train.yaml
+1 −1		recipes/stages/classification/train_multilabel.yaml