From eeddb833663b3f51cae977340f1a43ccb1d744c8 Mon Sep 17 00:00:00 2001 From: liuyuang Date: Tue, 7 Dec 2021 12:12:13 +0800 Subject: [PATCH] bug fix, test=allcase --- python/paddle/fluid/optimizer.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/python/paddle/fluid/optimizer.py b/python/paddle/fluid/optimizer.py index 7412d3a3fe6cf..556a59999282d 100755 --- a/python/paddle/fluid/optimizer.py +++ b/python/paddle/fluid/optimizer.py @@ -6582,8 +6582,16 @@ def mlp(input_x, input_y, hid_dim=128, label_dim=2): print("Finished apply_optimize") """ - return self._optimizer.apply_optimize( - loss, startup_program=startup_program, params_grads=params_grads) + if hasattr(self._optimizer, 'apply_optimize'): + return self._optimizer.apply_optimize( + loss, + startup_program=startup_program, + params_grads=params_grads) + elif hasattr(self._optimizer, '_apply_optimize'): + return self._optimizer._apply_optimize( + loss, + startup_program=startup_program, + params_grads=params_grads) def minimize(self, loss,