diff --git a/configs/ShuffleNet/ShuffleNetV2.yaml b/configs/ShuffleNet/ShuffleNetV2.yaml
index 1ee8787c0fd504b23c858416d643b4724933a6f5..c097afaea5a97bd02cdc5d3eef236a71a3feb4b7 100644
--- a/configs/ShuffleNet/ShuffleNetV2.yaml
+++ b/configs/ShuffleNet/ShuffleNetV2.yaml
@@ -14,7 +14,7 @@ topk: 5
 image_shape: [3, 224, 224]
 
 LEARNING_RATE:
-    function: 'Cosine'
+    function: 'CosineWarmup'
     params:
         lr: 0.5
         warmup_epoch: 5
diff --git a/configs/ShuffleNet/ShuffleNetV2_swish.yaml b/configs/ShuffleNet/ShuffleNetV2_swish.yaml
index 313f626fcd83491151a9496183290889e6e8e7dd..4e64ce8bb0562258c70234a2a7f888b0dbe08f8e 100644
--- a/configs/ShuffleNet/ShuffleNetV2_swish.yaml
+++ b/configs/ShuffleNet/ShuffleNetV2_swish.yaml
@@ -14,7 +14,7 @@ topk: 5
 image_shape: [3, 224, 224]
 
 LEARNING_RATE:
-    function: 'Cosine'
+    function: 'CosineWarmup'
     params:
         lr: 0.5
         warmup_epoch: 5
diff --git a/configs/ShuffleNet/ShuffleNetV2_x0_25.yaml b/configs/ShuffleNet/ShuffleNetV2_x0_25.yaml
index a8e8055e67da82e76fc9e46d30bcb075d832d66a..996f040bbd3fac8bd7d8bce672379cda647f0714 100644
--- a/configs/ShuffleNet/ShuffleNetV2_x0_25.yaml
+++ b/configs/ShuffleNet/ShuffleNetV2_x0_25.yaml
@@ -14,7 +14,7 @@ topk: 5
 image_shape: [3, 224, 224]
 
 LEARNING_RATE:
-    function: 'Cosine'
+    function: 'CosineWarmup'
     params:
         lr: 0.5
         warmup_epoch: 5
diff --git a/configs/ShuffleNet/ShuffleNetV2_x0_33.yaml b/configs/ShuffleNet/ShuffleNetV2_x0_33.yaml
index 9e1814013e99f9b1cc3236e4bf10c53386ec7321..f2941474105a4352ef1294711ecbb865a19b4774 100644
--- a/configs/ShuffleNet/ShuffleNetV2_x0_33.yaml
+++ b/configs/ShuffleNet/ShuffleNetV2_x0_33.yaml
@@ -14,7 +14,7 @@ topk: 5
 image_shape: [3, 224, 224]
 
 LEARNING_RATE:
-    function: 'Cosine'
+    function: 'CosineWarmup'
     params:
         lr: 0.5
         warmup_epoch: 5
diff --git a/configs/ShuffleNet/ShuffleNetV2_x0_5.yaml b/configs/ShuffleNet/ShuffleNetV2_x0_5.yaml
index be8f0be06396257490d4def0d906244a4150b0bd..05a1ad3eb94cb71e16e15eefb1bd195dee4b143b 100644
--- a/configs/ShuffleNet/ShuffleNetV2_x0_5.yaml
+++ b/configs/ShuffleNet/ShuffleNetV2_x0_5.yaml
@@ -14,7 +14,7 @@ topk: 5
 image_shape: [3, 224, 224]
 
 LEARNING_RATE:
-    function: 'Cosine'
+    function: 'CosineWarmup'
     params:
         lr: 0.5
         warmup_epoch: 5
diff --git a/configs/ShuffleNet/ShuffleNetV2_x1_5.yaml b/configs/ShuffleNet/ShuffleNetV2_x1_5.yaml
index a10ec37ca7af1afa830af25d3aa096a03f1febd1..63f50d48404e1bb0b9b9c599418df19465087a99 100644
--- a/configs/ShuffleNet/ShuffleNetV2_x1_5.yaml
+++ b/configs/ShuffleNet/ShuffleNetV2_x1_5.yaml
@@ -14,7 +14,7 @@ topk: 5
 image_shape: [3, 224, 224]
 
 LEARNING_RATE:
-    function: 'Cosine'
+    function: 'CosineWarmup'
     params:
         lr: 0.25
         warmup_epoch: 5
diff --git a/configs/ShuffleNet/ShuffleNetV2_x2_0.yaml b/configs/ShuffleNet/ShuffleNetV2_x2_0.yaml
index d84e29bc7075e73e44eb457cf32497c5b04671e2..5a14cebbf8e789a9e5c03d180f11bb0dc5a42e47 100644
--- a/configs/ShuffleNet/ShuffleNetV2_x2_0.yaml
+++ b/configs/ShuffleNet/ShuffleNetV2_x2_0.yaml
@@ -14,7 +14,7 @@ topk: 5
 image_shape: [3, 224, 224]
 
 LEARNING_RATE:
-    function: 'Cosine'
+    function: 'CosineWarmup'
     params:
         lr: 0.25
         warmup_epoch: 5