diff --git a/configs/MobileNetV3/MobileNetV3_large_x0_35.yaml b/configs/MobileNetV3/MobileNetV3_large_x0_35.yaml index 014f9e9b9..829f5375f 100644 --- a/configs/MobileNetV3/MobileNetV3_large_x0_35.yaml +++ b/configs/MobileNetV3/MobileNetV3_large_x0_35.yaml @@ -15,7 +15,7 @@ topk: 5 image_shape: [3, 224, 224] LEARNING_RATE: - function: 'Cosine' + function: 'CosineWarmup' params: lr: 2.6 warmup_epoch: 5 diff --git a/configs/MobileNetV3/MobileNetV3_large_x0_5.yaml b/configs/MobileNetV3/MobileNetV3_large_x0_5.yaml index e7d8726b7..9dab3b875 100644 --- a/configs/MobileNetV3/MobileNetV3_large_x0_5.yaml +++ b/configs/MobileNetV3/MobileNetV3_large_x0_5.yaml @@ -15,7 +15,7 @@ topk: 5 image_shape: [3, 224, 224] LEARNING_RATE: - function: 'Cosine' + function: 'CosineWarmup' params: lr: 1.3 warmup_epoch: 5 diff --git a/configs/MobileNetV3/MobileNetV3_large_x0_75.yaml b/configs/MobileNetV3/MobileNetV3_large_x0_75.yaml index c191c9b3c..124324d60 100644 --- a/configs/MobileNetV3/MobileNetV3_large_x0_75.yaml +++ b/configs/MobileNetV3/MobileNetV3_large_x0_75.yaml @@ -15,7 +15,7 @@ topk: 5 image_shape: [3, 224, 224] LEARNING_RATE: - function: 'Cosine' + function: 'CosineWarmup' params: lr: 1.3 warmup_epoch: 5 diff --git a/configs/MobileNetV3/MobileNetV3_large_x1_0.yaml b/configs/MobileNetV3/MobileNetV3_large_x1_0.yaml index 6e90d5590..9c85f44fa 100644 --- a/configs/MobileNetV3/MobileNetV3_large_x1_0.yaml +++ b/configs/MobileNetV3/MobileNetV3_large_x1_0.yaml @@ -15,7 +15,7 @@ topk: 5 image_shape: [3, 224, 224] LEARNING_RATE: - function: 'Cosine' + function: 'CosineWarmup' params: lr: 1.3 warmup_epoch: 5 diff --git a/configs/MobileNetV3/MobileNetV3_large_x1_25.yaml b/configs/MobileNetV3/MobileNetV3_large_x1_25.yaml index 59f3b8f2c..3f5f48bbc 100644 --- a/configs/MobileNetV3/MobileNetV3_large_x1_25.yaml +++ b/configs/MobileNetV3/MobileNetV3_large_x1_25.yaml @@ -15,7 +15,7 @@ topk: 5 image_shape: [3, 224, 224] LEARNING_RATE: - function: 'Cosine' + function: 'CosineWarmup' params: lr: 0.65 warmup_epoch: 5 diff --git a/configs/MobileNetV3/MobileNetV3_small_x0_35.yaml b/configs/MobileNetV3/MobileNetV3_small_x0_35.yaml index a9b4d4322..ff5dcd714 100644 --- a/configs/MobileNetV3/MobileNetV3_small_x0_35.yaml +++ b/configs/MobileNetV3/MobileNetV3_small_x0_35.yaml @@ -14,7 +14,7 @@ topk: 5 image_shape: [3, 224, 224] LEARNING_RATE: - function: 'Cosine' + function: 'CosineWarmup' params: lr: 2.6 warmup_epoch: 5 diff --git a/configs/MobileNetV3/MobileNetV3_small_x0_5.yaml b/configs/MobileNetV3/MobileNetV3_small_x0_5.yaml index 95520fdb2..af0af7c10 100644 --- a/configs/MobileNetV3/MobileNetV3_small_x0_5.yaml +++ b/configs/MobileNetV3/MobileNetV3_small_x0_5.yaml @@ -15,7 +15,7 @@ topk: 5 image_shape: [3, 224, 224] LEARNING_RATE: - function: 'Cosine' + function: 'CosineWarmup' params: lr: 2.6 warmup_epoch: 5 diff --git a/configs/MobileNetV3/MobileNetV3_small_x0_75.yaml b/configs/MobileNetV3/MobileNetV3_small_x0_75.yaml index dc20e57b3..4eae9a92a 100644 --- a/configs/MobileNetV3/MobileNetV3_small_x0_75.yaml +++ b/configs/MobileNetV3/MobileNetV3_small_x0_75.yaml @@ -15,7 +15,7 @@ topk: 5 image_shape: [3, 224, 224] LEARNING_RATE: - function: 'Cosine' + function: 'CosineWarmup' params: lr: 2.6 warmup_epoch: 5 diff --git a/configs/MobileNetV3/MobileNetV3_small_x1_0.yaml b/configs/MobileNetV3/MobileNetV3_small_x1_0.yaml index 9c315071a..f63ce6d92 100644 --- a/configs/MobileNetV3/MobileNetV3_small_x1_0.yaml +++ b/configs/MobileNetV3/MobileNetV3_small_x1_0.yaml @@ -15,7 +15,7 @@ topk: 5 image_shape: [3, 224, 224] LEARNING_RATE: - function: 'Cosine' + function: 'CosineWarmup' params: lr: 2.6 warmup_epoch: 5 diff --git a/configs/MobileNetV3/MobileNetV3_small_x1_25.yaml b/configs/MobileNetV3/MobileNetV3_small_x1_25.yaml index 80b53901a..bc7b3193d 100644 --- a/configs/MobileNetV3/MobileNetV3_small_x1_25.yaml +++ b/configs/MobileNetV3/MobileNetV3_small_x1_25.yaml @@ -15,7 +15,7 @@ topk: 5 image_shape: [3, 224, 224] LEARNING_RATE: - function: 'Cosine' + function: 'CosineWarmup' params: lr: 1.3 warmup_epoch: 5