Skip to content

Commit 300c539

Browse files
committed
update: MADGRAD optimizer
1 parent 04b22d5 commit 300c539

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

pytorch_optimizer/optimizer/madgrad.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -101,7 +101,6 @@ def step(self, closure: CLOSURE = None) -> LOSS:
101101
raise NoSparseGradientError(str(self), note='momentum > 0.0')
102102

103103
grad_sum_sq, s = state['grad_sum_sq'], state['s']
104-
105104
if weight_decay > 0.0 and not self.decouple_decay:
106105
if grad.is_sparse:
107106
raise NoSparseGradientError(str(self), note='weight_decay')

0 commit comments

Comments
 (0)