Skip to content

Commit 1e67fa9

Browse files
committed
fix: typo
1 parent b18d662 commit 1e67fa9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pytorch_optimizer/adan.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -112,7 +112,7 @@ def step(self, closure: CLOSURE = None) -> LOSS:
112112
state['previous_grad'] = grad.clone()
113113

114114
exp_avg.mul_(beta1).add_(grad, alpha=1.0 - beta1)
115-
exp_avg_var.mul_(beta2).add_(grad_diff, value=1.0 - beta2)
115+
exp_avg_var.mul_(beta2).add_(grad_diff, alpha=1.0 - beta2)
116116
exp_avg_nest.mul_(beta3).add_((grad + beta2 * grad_diff) ** 2, alpha=1.0 - beta3)
117117

118118
step_size = group['lr'] / math.sqrt(exp_avg_nest + self.eps)

0 commit comments

Comments
 (0)