From 2959783b54b7ee0f4695c23f76dd2c24e479200d Mon Sep 17 00:00:00 2001
From: LielinJiang <jianglielin@baidu.com>
Date: Thu, 15 Oct 2020 06:41:33 +0000
Subject: [PATCH] fix lr scheduler

---
 ppgan/solver/lr_scheduler.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/ppgan/solver/lr_scheduler.py b/ppgan/solver/lr_scheduler.py
index 8e3f87b..39b6389 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
-- 
GitLab