Skip to content

Commit 840429a

Browse files
Update AdamW.py
1 parent 25350da commit 840429a

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/AdamW.py

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

297-
_, _, _, _, _, _ = paddle._C_ops.adamw_(
297+
_, _, _, _, _, *_ = paddle._C_ops.adamw_(
298298
param_,
299299
grad_,
300300
lr,

0 commit comments

Comments
 (0)