We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f2f660d commit 0f8b1e6Copy full SHA for 0f8b1e6
pytorch_optimizer/madgrad.py
@@ -59,8 +59,6 @@ def check_valid_parameters(self):
59
raise ValueError(f'Invalid learning rate : {self.lr}')
60
if self.weight_decay < 0.0:
61
raise ValueError(f'Invalid weight_decay : {self.weight_decay}')
62
- if not 0.0 < self.momentum <= 1.0:
63
- raise ValueError(f'Invalid momentum : {self.momentum}')
64
if self.eps < 0.0:
65
raise ValueError(f'Invalid eps : {self.eps}')
66
0 commit comments