fix bug of import

PaddlePaddle · Jun 20, 2023 · 26a1caf · 26a1caf
1 parent f327122
commit 26a1caf
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/python/paddle/distributed/fleet/meta_optimizers/lamb_optimizer.py b/python/paddle/distributed/fleet/meta_optimizers/lamb_optimizer.py
@@ -110,7 +110,7 @@ def apply_gradients(self, params_grads):
         return self.lamb_opt.apply_gradients(params_grads=params_grads)
 
     def apply_optimize(self, loss, startup_program, params_grads):
-        return self.lamb_opt.apply_optimize(
+        return self.lamb_opt._apply_optimize(
             loss, startup_program=startup_program, params_grads=params_grads
         )
 

diff --git a/test/amp/test_amp_api.py b/test/amp/test_amp_api.py
@@ -70,7 +70,7 @@ def check_results(
                 )
                 out = model(x)
                 loss = paddle.mean(out)
-                optimizer = paddle.fluid.optimizer.Adadelta(learning_rate=0.001)
+                optimizer = paddle.optimizer.Adadelta(learning_rate=0.001)
                 optimizer = paddle.static.amp.decorate(
                     optimizer,
                     init_loss_scaling=128.0,