Skip to content

Commit

Permalink
-
Browse files Browse the repository at this point in the history
  • Loading branch information
matsen committed Jul 9, 2024
1 parent 7b737a7 commit c7bab26
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion netam/framework.py
Original file line number Diff line number Diff line change
Expand Up @@ -421,7 +421,7 @@ def reset_optimization(self, learning_rate=None):
if learning_rate is None:
learning_rate = self.learning_rate

# copied from # https://github.com/karpathy/nanoGPT/blob/9755682b981a45507f6eb9b11eadef8cb83cebd5/model.py#L264
# copied from https://github.com/karpathy/nanoGPT/blob/9755682b981a45507f6eb9b11eadef8cb83cebd5/model.py#L264
param_dict = {
pn: p for pn, p in self.model.named_parameters() if p.requires_grad
}
Expand Down

0 comments on commit c7bab26

Please sign in to comment.