bug fix for adamw (#37905)

9a2d327c · Yuang Liu · GitHub · b8793f70 · 9a2d327c
隐藏空白更改
内联并排

Showing with 4 addition and 1 deletion

python/paddle/fluid/optimizer.py python/paddle/fluid/optimizer.py +4 -1

未找到文件。
--- a/python/paddle/fluid/optimizer.py
+++ b/python/paddle/fluid/optimizer.py
@@ -6582,7 +6582,10 @@ class RecomputeOptimizer(Optimizer):
                print("Finished apply_optimize")
        """
-        return self._optimizer.apply_optimize(
+        func = self._optimizer.apply_optimize if hasattr(
+            self._optimizer,
+            'apply_optimize') else self._optimizer._apply_optimize
+        return func(
            loss, startup_program=startup_program, params_grads=params_grads)
    def minimize(self,