Skip to content

Commit b3896a5

Browse files
committed
[skip ci] refactor: remove comment
1 parent 48342a5 commit b3896a5

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

pytorch_optimizer/adahessian.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -187,9 +187,7 @@ def step(self, closure: CLOSURE = None) -> LOSS:
187187

188188
if len(state) == 1:
189189
state['step'] = 0
190-
# Exponential moving average of gradient values
191190
state['exp_avg'] = torch.zeros_like(p.data)
192-
# Exponential moving average of Hessian diagonal square values
193191
state['exp_hessian_diag_sq'] = torch.zeros_like(p.data)
194192

195193
exp_avg, exp_hessian_diag_sq = (

0 commit comments

Comments
 (0)