Fix untest error in test_adamax_op (#51692)

PaddlePaddle · Mar 17, 2023 · d9d3592 · d9d3592
1 parent 3b2cd23
commit d9d3592
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/python/paddle/fluid/optimizer.py b/python/paddle/fluid/optimizer.py
@@ -2809,7 +2809,7 @@ def _finish_update(self, block, parameters_and_grads):
             with param.block.program._optimized_guard(
                 [param, grad]
             ), name_scope('adamx'):
-                beta1_pow_acc = self._get_accumulator(
+                beta1_pow_acc = self._get_accumulator_master(
                     self._beta1_pow_acc_str, param
                 )
                 if in_dygraph_mode():

diff --git a/python/paddle/fluid/tests/unittests/test_adamax_op.py b/python/paddle/fluid/tests/unittests/test_adamax_op.py
@@ -31,6 +31,7 @@ def adamx_wrapper(
     beta1=0.78,
     beta2=0.899,
     epsilon=1e-5,
+    find_master=False,
 ):
     return paddle._C_ops.adamax_(
         param,
@@ -43,6 +44,7 @@ def adamx_wrapper(
         beta1,
         beta2,
         epsilon,
+        find_master,
     )