diff --git a/ppgan/solver/lr_scheduler.py b/ppgan/solver/lr_scheduler.py
index 8e3f87baa3a76d31bfdd06c1646131d47dedeffa..39b6389ebd3ea09186ee1d04e497a8bc27a3165b 100644
--- a/ppgan/solver/lr_scheduler.py
+++ b/ppgan/solver/lr_scheduler.py
@@ -12,8 +12,8 @@ def build_lr_scheduler(cfg):
                 0, epoch + 1 - cfg.start_epoch) / float(cfg.decay_epochs + 1)
             return lr_l
 
-        scheduler = paddle.optimizer.lr.LambdaLR(cfg.learning_rate,
-                                                 lr_lambda=lambda_rule)
+        scheduler = paddle.optimizer.lr.LambdaDecay(cfg.learning_rate,
+                                                    lr_lambda=lambda_rule)
         return scheduler
     else:
         raise NotImplementedError