瀏覽代碼

Merge pull request #1357 from FlyingQianMM/develop_ui

fix cosine learning rate setting error for restful classification
FlyingQianMM 3 年之前
父節點
當前提交
eb4a388e24
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      paddlex_restful/restful/project/train/classification.py

+ 1 - 1
paddlex_restful/restful/project/train/classification.py

@@ -70,7 +70,7 @@ def build_optimizer(parameters, step_each_epoch, params):
     num_epochs = params.num_epochs
     if params.lr_policy == 'Cosine':
         learning_rate = paddle.optimizer.lr.CosineAnnealingDecay(
-            learning_rate=.001, T_max=step_each_epoch * num_epochs)
+            learning_rate=learning_rate, T_max=step_each_epoch * num_epochs)
     elif params.lr_policy == 'Linear':
         learning_rate = paddle.optimizer.lr.PolynomialDecay(
             learning_rate=learning_rate,