From 15168d25f306b5c69cc81495676cb1a931ce5c83 Mon Sep 17 00:00:00 2001
From: littletomatodonkey <2120160898@bit.edu.cn>
Date: Wed, 19 May 2021 12:49:19 +0800
Subject: [PATCH] fix mv3 cosine warmup (#714)

---
 configs/MobileNetV3/MobileNetV3_large_x0_35.yaml | 2 +-
 configs/MobileNetV3/MobileNetV3_large_x0_5.yaml  | 2 +-
 configs/MobileNetV3/MobileNetV3_large_x0_75.yaml | 2 +-
 configs/MobileNetV3/MobileNetV3_large_x1_0.yaml  | 2 +-
 configs/MobileNetV3/MobileNetV3_large_x1_25.yaml | 2 +-
 configs/MobileNetV3/MobileNetV3_small_x0_35.yaml | 2 +-
 configs/MobileNetV3/MobileNetV3_small_x0_5.yaml  | 2 +-
 configs/MobileNetV3/MobileNetV3_small_x0_75.yaml | 2 +-
 configs/MobileNetV3/MobileNetV3_small_x1_0.yaml  | 2 +-
 configs/MobileNetV3/MobileNetV3_small_x1_25.yaml | 2 +-
 10 files changed, 10 insertions(+), 10 deletions(-)

diff --git a/configs/MobileNetV3/MobileNetV3_large_x0_35.yaml b/configs/MobileNetV3/MobileNetV3_large_x0_35.yaml
index 014f9e9b..829f5375 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 e7d8726b..9dab3b87 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 c191c9b3..124324d6 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 6e90d559..9c85f44f 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 59f3b8f2..3f5f48bb 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 a9b4d432..ff5dcd71 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 95520fdb..af0af7c1 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 dc20e57b..4eae9a92 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 9c315071..f63ce6d9 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 80b53901..bc7b3193 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
-- 
GitLab