Skip to content

Commit 2162e68

Browse files
committed
update: missing check_valid_parameters()
1 parent 62f518b commit 2162e68

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

pytorch_optimizer/pcgrad.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,8 @@ def __init__(self, optimizer: Optimizer, reduction: str = 'mean'):
3030
self.optimizer = optimizer
3131
self.reduction = reduction
3232

33+
self.check_valid_parameters()
34+
3335
def check_valid_parameters(self):
3436
if self.reduction not in ('mean', 'sum'):
3537
raise ValueError(f'invalid reduction : {self.reduction}')

0 commit comments

Comments
 (0)