Explorar o código

Merge pull request #1357 from FlyingQianMM/develop_ui

fix cosine learning rate setting error for restful classification
FlyingQianMM %!s(int64=3) %!d(string=hai) anos
pai
achega
eb4a388e24
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  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,