fix squeezenet (#348)

bcda9c2b · littletomatodonkey · GitHub · 7ef47416 · bcda9c2b
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

configs/SqueezeNet/SqueezeNet1_0.yaml configs/SqueezeNet/SqueezeNet1_0.yaml +1 -1

未找到文件。
--- 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