From bcda9c2b1fea857e0aa9fbca3bc53e2826e83dc9 Mon Sep 17 00:00:00 2001
From: littletomatodonkey <2120160898@bit.edu.cn>
Date: Fri, 30 Oct 2020 13:36:08 +0800
Subject: [PATCH] fix squeezenet (#348)

---
 configs/SqueezeNet/SqueezeNet1_0.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/configs/SqueezeNet/SqueezeNet1_0.yaml b/configs/SqueezeNet/SqueezeNet1_0.yaml
index 9107bdfe..9809ba11 100644
--- a/configs/SqueezeNet/SqueezeNet1_0.yaml
+++ b/configs/SqueezeNet/SqueezeNet1_0.yaml
@@ -14,7 +14,7 @@ topk: 5
 image_shape: [3, 224, 224]
 
 LEARNING_RATE:
-   function: 'Cosine'
+   function: 'CosineWarmup'
    params:
        lr: 0.02
 
-- 
GitLab