Skip to content

Commit 17893ed

Browse files
authored
Merge pull request #238 from michaldyczko/237-fix
fix: change Lookahead's param_groups to a property
2 parents 7205a27 + 25b4f20 commit 17893ed

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

pytorch_optimizer/optimizer/lookahead.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,6 @@ def __init__(
3636
self.pullback_momentum = pullback_momentum
3737

3838
self.optimizer = optimizer
39-
self.param_groups = self.optimizer.param_groups
4039

4140
self.state: STATE = defaultdict(dict)
4241

@@ -58,6 +57,10 @@ def __init__(
5857
**optimizer.defaults,
5958
}
6059

60+
@property
61+
def param_groups(self):
62+
return self.optimizer.param_groups
63+
6164
def __getstate__(self):
6265
return {
6366
'state': self.state,

0 commit comments

Comments
 (0)