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 dece8c0 commit 979cd5cCopy full SHA for 979cd5c
pytorch_optimizer/optimizer/prodigy.py
@@ -47,7 +47,7 @@ def __init__(
47
eps: float = 1e-8,
48
):
49
self.validate_learning_rate(lr)
50
- self.validate_betas(betas)
+ self.validate_betas((*betas, beta3))
51
self.validate_non_negative(weight_decay, 'weight_decay')
52
self.validate_non_negative(eps, 'eps')
53
0 commit comments