Skip to content

Commit cf5ee0d

Browse files
committed
refactor: p.grad to grad
1 parent 2890bd4 commit cf5ee0d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pytorch_optimizer/optimizer/adasmooth.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,7 @@ def step(self, closure: CLOSURE = None) -> LOSS:
9292

9393
self.apply_weight_decay(
9494
p=p,
95-
grad=p.grad,
95+
grad=grad,
9696
lr=group['lr'],
9797
weight_decay=group['weight_decay'],
9898
weight_decouple=group['weight_decouple'],

0 commit comments

Comments
 (0)