Skip to content

Commit 979cd5c

Browse files
committed
update: validate_betas
1 parent dece8c0 commit 979cd5c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pytorch_optimizer/optimizer/prodigy.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ def __init__(
4747
eps: float = 1e-8,
4848
):
4949
self.validate_learning_rate(lr)
50-
self.validate_betas(betas)
50+
self.validate_betas((*betas, beta3))
5151
self.validate_non_negative(weight_decay, 'weight_decay')
5252
self.validate_non_negative(eps, 'eps')
5353

0 commit comments

Comments
 (0)