update hpo_config.yaml (#1240)

openvinotoolkit · Sep 5, 2022 · 5f622b4 · 5f622b4
1 parent bc90480
commit 5f622b4
Show file tree

Hide file tree

Showing 9 changed files with 36 additions and 38 deletions.
diff --git a/...del-preparation-algorithm/configs/classification/efficientnet_b0_cls_incr/hpo_config.yaml b/...del-preparation-algorithm/configs/classification/efficientnet_b0_cls_incr/hpo_config.yaml
@@ -4,12 +4,12 @@ hp_space:
   learning_parameters.learning_rate:
     param_type: qloguniform
     range:
-      - 0.0003
-      - 0.1
+      - 0.0014
+      - 0.035
       - 0.0001
   learning_parameters.batch_size:
     param_type: qloguniform
     range:
-      - 64
-      - 256
+      - 42
+      - 96
       - 2
diff --git a/...l-preparation-algorithm/configs/classification/efficientnet_v2_s_cls_incr/hpo_config.yaml b/...l-preparation-algorithm/configs/classification/efficientnet_v2_s_cls_incr/hpo_config.yaml
@@ -4,12 +4,12 @@ hp_space:
   learning_parameters.learning_rate:
     param_type: qloguniform
     range:
-      - 0.0014
-      - 0.035
+      - 0.0007
+      - 0.07
       - 0.0001
   learning_parameters.batch_size:
     param_type: qloguniform
     range:
-      - 40
-      - 96
+      - 32
+      - 128
       - 2
diff --git a/...reparation-algorithm/configs/classification/mobilenet_v3_large_1_cls_incr/hpo_config.yaml b/...reparation-algorithm/configs/classification/mobilenet_v3_large_1_cls_incr/hpo_config.yaml
@@ -4,12 +4,12 @@ hp_space:
   learning_parameters.learning_rate:
     param_type: qloguniform
     range:
-      - 0.0032
-      - 0.08
+      - 0.0008
+      - 0.1
       - 0.0001
   learning_parameters.batch_size:
     param_type: qloguniform
     range:
-      - 40
-      - 96
+      - 64
+      - 256
       - 2
diff --git a/...l/model-preparation-algorithm/configs/detection/cspdarknet_yolox_cls_incr/hpo_config.yaml b/...l/model-preparation-algorithm/configs/detection/cspdarknet_yolox_cls_incr/hpo_config.yaml
@@ -5,12 +5,12 @@ hp_space:
   learning_parameters.learning_rate:
     param_type: qloguniform
     range:
-      - 0.00004
-      - 0.001
+      - 0.00002
+      - 0.002
       - 0.00001
   learning_parameters.batch_size:
     param_type: qloguniform
     range:
       - 4
-      - 32
+      - 16
       - 2
diff --git a/...l/model-preparation-algorithm/configs/detection/mobilenetv2_atss_cls_incr/hpo_config.yaml b/...l/model-preparation-algorithm/configs/detection/mobilenetv2_atss_cls_incr/hpo_config.yaml
@@ -12,5 +12,5 @@ hp_space:
     param_type: qloguniform
     range:
       - 4
-      - 32
+      - 16
       - 2
diff --git a/...al/model-preparation-algorithm/configs/detection/mobilenetv2_ssd_cls_incr/hpo_config.yaml b/...al/model-preparation-algorithm/configs/detection/mobilenetv2_ssd_cls_incr/hpo_config.yaml
@@ -5,12 +5,12 @@ hp_space:
   learning_parameters.learning_rate:
     param_type: qloguniform
     range:
-      - 0.001
-      - 0.1
+      - 0.002
+      - 0.05
       - 0.001
   learning_parameters.batch_size:
     param_type: qloguniform
     range:
-      - 4
-      - 32
+      - 6
+      - 12
       - 2
diff --git a/...l/model-preparation-algorithm/configs/segmentation/ocr-lite-hrnet-18-mod2/hpo_config.yaml b/...l/model-preparation-algorithm/configs/segmentation/ocr-lite-hrnet-18-mod2/hpo_config.yaml
@@ -2,14 +2,14 @@ metric: mDice
 search_algorithm: asha
 hp_space:
   learning_parameters.learning_rate:
-    param_type: quniform
+    param_type: qloguniform
     range:
-      - 0.001
-      - 0.1
-      - 0.001
+      - 0.0002
+      - 0.005
+      - 0.0001
   learning_parameters.batch_size:
     param_type: qloguniform
     range:
-      - 4
-      - 16
+      - 6
+      - 12
       - 2
diff --git a/...al/model-preparation-algorithm/configs/segmentation/ocr-lite-hrnet-s-mod2/hpo_config.yaml b/...al/model-preparation-algorithm/configs/segmentation/ocr-lite-hrnet-s-mod2/hpo_config.yaml
@@ -1,16 +1,15 @@
 metric: mDice
 search_algorithm: asha
-# max_iterations: 30
 hp_space:
   learning_parameters.learning_rate:
-    param_type: quniform
+    param_type: qloguniform
     range:
-      - 0.001
-      - 0.1
-      - 0.001
+      - 0.0002
+      - 0.005
+      - 0.0001
   learning_parameters.batch_size:
     param_type: qloguniform
     range:
-      - 4
-      - 16
+      - 6
+      - 12
       - 2
diff --git a/...al/model-preparation-algorithm/configs/segmentation/ocr-lite-hrnet-x-mod3/hpo_config.yaml b/...al/model-preparation-algorithm/configs/segmentation/ocr-lite-hrnet-x-mod3/hpo_config.yaml
@@ -1,13 +1,12 @@
 metric: mDice
 search_algorithm: asha
-# max_iterations: 30
 hp_space:
   learning_parameters.learning_rate:
-    param_type: quniform
+    param_type: qloguniform
     range:
-      - 0.001
-      - 0.1
-      - 0.001
+      - 0.0001
+      - 0.01
+      - 0.0001
   learning_parameters.batch_size:
     param_type: qloguniform
     range:
-Original file line number
+Diff line change
@@ Expand Up / @@ -12,5 +12,5 @@ hp_space: @@
         param_type: qloguniform
         range:
           - 4
-          - 32
+          - 16
           - 2