Просмотр исходного кода

fix cosine learning rate setting error for restful classification

FlyingQianMM 3 лет назад
Родитель
Сommit
41aba5f820
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,