Skip to content

Commit 25350da

Browse files
Update Adam.py
1 parent 9b65d57 commit 25350da

File tree

1 file changed

+1
-1
lines changed
  • backends/sdaa/sdaa_ext/python/custom_parallel

1 file changed

+1
-1
lines changed

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -292,7 +292,7 @@ def _append_optimize_op(self, block, param_and_grad):
292292
moment1_ = moment1
293293
moment2_ = moment2
294294

295-
_, _, _, _, _, _ = paddle._C_ops.adam_(
295+
_, _, _, _, _, *_ = paddle._C_ops.adam_(
296296
param_,
297297
grad_,
298298
lr,

0 commit comments

Comments
 (0)