fix bug of import

longranger2 · longranger2 · commit 26a1cafc8531 · 2023-06-20T02:19:47.000Z
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,

Original file line number	Diff line number	Diff line change
`@@ -110,7 +110,7 @@ def apply_gradients(self, params_grads):`
`110`	`110`	`return self.lamb_opt.apply_gradients(params_grads=params_grads)`
`111`	`111`
`112`	`112`	`def apply_optimize(self, loss, startup_program, params_grads):`
`113`		`- return self.lamb_opt.apply_optimize(`
	`113`	`+ return self.lamb_opt._apply_optimize(`
`114`	`114`	`loss, startup_program=startup_program, params_grads=params_grads`
`115`	`115`	`)`
`116`	`116`
Original file line number	Diff line number	Diff line change
`@@ -70,7 +70,7 @@ def check_results(`
`70`	`70`	`)`
`71`	`71`	`out = model(x)`
`72`	`72`	`loss = paddle.mean(out)`
`73`		`- optimizer = paddle.fluid.optimizer.Adadelta(learning_rate=0.001)`
	`73`	`+ optimizer = paddle.optimizer.Adadelta(learning_rate=0.001)`
`74`	`74`	`optimizer = paddle.static.amp.decorate(`
`75`	`75`	`optimizer,`
`76`	`76`	`init_loss_scaling=128.0,`