From 0fd505516a161fbc4e7673e1d9f25b637e8029f4 Mon Sep 17 00:00:00 2001
From: LoneRanger <836253168@qq.com>
Date: Mon, 3 Jul 2023 11:38:59 +0800
Subject: [PATCH] fix the example code (#55053)

---
 python/paddle/fluid/optimizer.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/python/paddle/fluid/optimizer.py b/python/paddle/fluid/optimizer.py
index 87b486c2188..113170ae59d 100755
--- a/python/paddle/fluid/optimizer.py
+++ b/python/paddle/fluid/optimizer.py
@@ -580,17 +580,18 @@ class Optimizer:
 
                     bd = [2, 4, 6, 8]
                     value = [0.2, 0.4, 0.6, 0.8, 1.0]
-                    adam = fluid.optimizer.Adam(fluid.dygraph.PiecewiseDecay(bd, value, 0),
-                                           parameter_list=linear.parameters())
+                    adam = paddle.optimizer.Adam(paddle.optimizer.lr.PiecewiseDecay(bd, value),
+                                           parameters=linear.parameters())
 
                     # first step: learning rate is 0.2
-                    np.allclose(adam.current_step_lr(), 0.2, rtol=1e-06, atol=0.0) # True
+                    np.allclose(adam.get_lr(), 0.2, rtol=1e-06, atol=0.0) # True
 
                     # learning rate for different steps
                     ret = [0.2, 0.2, 0.4, 0.4, 0.6, 0.6, 0.8, 0.8, 1.0, 1.0, 1.0, 1.0]
                     for i in range(12):
                         adam.minimize(loss)
-                        lr = adam.current_step_lr()
+                        adam.step()
+                        lr = adam.get_lr()
                         np.allclose(lr, ret[i], rtol=1e-06, atol=0.0) # True
 
         """
-- 
GitLab