bug fix, test=allcase

PaddlePaddle · Dec 7, 2021 · eeddb83 · eeddb83
1 parent 4035bd2
commit eeddb83
Showing 1 changed file with 10 additions and 2 deletions.
diff --git 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,