restorecode

Author: HydrogenSulfate

backends/sdaa/sdaa_ext/python/custom_parallel/Adam.py

@@ -292,7 +292,7 @@ def _append_optimize_op(self, block, param_and_grad):
             moment1_ = moment1
             moment2_ = moment2
 
-        _, _, _, _, _, *_ = paddle._C_ops.adam_(
+        _, _, _, _, _, _ = paddle._C_ops.adam_(
             param_,
             grad_,
             lr,

backends/sdaa/sdaa_ext/python/custom_parallel/AdamW.py

@@ -294,7 +294,7 @@ def _append_optimize_op(self, block, param_and_grad):
             moment1_ = moment1
             moment2_ = moment2
 
-        _, _, _, _, _, *_ = paddle._C_ops.adamw_(
+        _, _, _, _, _, _ = paddle._C_ops.adamw_(
             param_,
             grad_,
             lr,