From 4cdcd4af7435b3a62dbd2eb70a26ee92ac4fa8b6 Mon Sep 17 00:00:00 2001
From: weishengyu <weishengyu@baidu.com>
Date: Thu, 22 Oct 2020 16:11:14 +0800
Subject: [PATCH] shufflenet lr cosine -> cosine warmup

---
 configs/ShuffleNet/ShuffleNetV2.yaml       | 2 +-
 configs/ShuffleNet/ShuffleNetV2_swish.yaml | 2 +-
 configs/ShuffleNet/ShuffleNetV2_x0_25.yaml | 2 +-
 configs/ShuffleNet/ShuffleNetV2_x0_33.yaml | 2 +-
 configs/ShuffleNet/ShuffleNetV2_x0_5.yaml  | 2 +-
 configs/ShuffleNet/ShuffleNetV2_x1_5.yaml  | 2 +-
 configs/ShuffleNet/ShuffleNetV2_x2_0.yaml  | 2 +-
 7 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/configs/ShuffleNet/ShuffleNetV2.yaml b/configs/ShuffleNet/ShuffleNetV2.yaml
index 1ee8787c..c097afae 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 313f626f..4e64ce8b 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 a8e8055e..996f040b 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 9e181401..f2941474 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 be8f0be0..05a1ad3e 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 a10ec37c..63f50d48 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 d84e29bc..5a14cebb 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
-- 
GitLab