From 5f622b42c2f75266f8eb6bb868d542fcc950982e Mon Sep 17 00:00:00 2001
From: Eunwoo Shin <eunwoo.shin@intel.com>
Date: Mon, 5 Sep 2022 11:24:24 +0900
Subject: [PATCH] update hpo_config.yaml (#1240)
---
.../efficientnet_b0_cls_incr/hpo_config.yaml | 8 ++++----
.../efficientnet_v2_s_cls_incr/hpo_config.yaml | 8 ++++----
.../mobilenet_v3_large_1_cls_incr/hpo_config.yaml | 8 ++++----
.../cspdarknet_yolox_cls_incr/hpo_config.yaml | 6 +++---
.../mobilenetv2_atss_cls_incr/hpo_config.yaml | 2 +-
.../mobilenetv2_ssd_cls_incr/hpo_config.yaml | 8 ++++----
.../ocr-lite-hrnet-18-mod2/hpo_config.yaml | 12 ++++++------
.../ocr-lite-hrnet-s-mod2/hpo_config.yaml | 13 ++++++-------
.../ocr-lite-hrnet-x-mod3/hpo_config.yaml | 9 ++++-----
9 files changed, 36 insertions(+), 38 deletions(-)
diff --git a/external/model-preparation-algorithm/configs/classification/efficientnet_b0_cls_incr/hpo_config.yaml b/external/model-preparation-algorithm/configs/classification/efficientnet_b0_cls_incr/hpo_config.yaml
index ba221b39335..5ea8a9b33a0 100644
--- a/external/model-preparation-algorithm/configs/classification/efficientnet_b0_cls_incr/hpo_config.yaml
+++ b/external/model-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/external/model-preparation-algorithm/configs/classification/efficientnet_v2_s_cls_incr/hpo_config.yaml b/external/model-preparation-algorithm/configs/classification/efficientnet_v2_s_cls_incr/hpo_config.yaml
index e5b6ddbd7d0..4b79b24194c 100644
--- a/external/model-preparation-algorithm/configs/classification/efficientnet_v2_s_cls_incr/hpo_config.yaml
+++ b/external/model-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/external/model-preparation-algorithm/configs/classification/mobilenet_v3_large_1_cls_incr/hpo_config.yaml b/external/model-preparation-algorithm/configs/classification/mobilenet_v3_large_1_cls_incr/hpo_config.yaml
index 58384b4c619..35a77712ea5 100644
--- a/external/model-preparation-algorithm/configs/classification/mobilenet_v3_large_1_cls_incr/hpo_config.yaml
+++ b/external/model-preparation-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/external/model-preparation-algorithm/configs/detection/cspdarknet_yolox_cls_incr/hpo_config.yaml b/external/model-preparation-algorithm/configs/detection/cspdarknet_yolox_cls_incr/hpo_config.yaml
index fffaa83d8c5..a89872d9b69 100644
--- a/external/model-preparation-algorithm/configs/detection/cspdarknet_yolox_cls_incr/hpo_config.yaml
+++ b/external/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/external/model-preparation-algorithm/configs/detection/mobilenetv2_atss_cls_incr/hpo_config.yaml b/external/model-preparation-algorithm/configs/detection/mobilenetv2_atss_cls_incr/hpo_config.yaml
index 1ffda84eac4..8df3d6be9dd 100644
--- a/external/model-preparation-algorithm/configs/detection/mobilenetv2_atss_cls_incr/hpo_config.yaml
+++ b/external/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/external/model-preparation-algorithm/configs/detection/mobilenetv2_ssd_cls_incr/hpo_config.yaml b/external/model-preparation-algorithm/configs/detection/mobilenetv2_ssd_cls_incr/hpo_config.yaml
index eebed24be1e..8a5b9ff69fa 100644
--- a/external/model-preparation-algorithm/configs/detection/mobilenetv2_ssd_cls_incr/hpo_config.yaml
+++ b/external/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/external/model-preparation-algorithm/configs/segmentation/ocr-lite-hrnet-18-mod2/hpo_config.yaml b/external/model-preparation-algorithm/configs/segmentation/ocr-lite-hrnet-18-mod2/hpo_config.yaml
index abec751c13c..819a3ac0368 100644
--- a/external/model-preparation-algorithm/configs/segmentation/ocr-lite-hrnet-18-mod2/hpo_config.yaml
+++ b/external/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/external/model-preparation-algorithm/configs/segmentation/ocr-lite-hrnet-s-mod2/hpo_config.yaml b/external/model-preparation-algorithm/configs/segmentation/ocr-lite-hrnet-s-mod2/hpo_config.yaml
index 01d052bd2f1..819a3ac0368 100644
--- a/external/model-preparation-algorithm/configs/segmentation/ocr-lite-hrnet-s-mod2/hpo_config.yaml
+++ b/external/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/external/model-preparation-algorithm/configs/segmentation/ocr-lite-hrnet-x-mod3/hpo_config.yaml b/external/model-preparation-algorithm/configs/segmentation/ocr-lite-hrnet-x-mod3/hpo_config.yaml
index 01d052bd2f1..3b8a4838ecc 100644
--- a/external/model-preparation-algorithm/configs/segmentation/ocr-lite-hrnet-x-mod3/hpo_config.yaml
+++ b/external/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: