Skip to content

Commit

Permalink
Merge pull request #299 from AIM-Harvard/version_fix
Browse files Browse the repository at this point in the history
Safety fixes
  • Loading branch information
surajpaib authored May 15, 2024
2 parents 31c1e3d + f0c66ea commit 52c3769
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions fmcib/optimizers/lars.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
- https://arxiv.org/pdf/1708.03888.pdf
- https://github.com/pytorch/pytorch/blob/1.6/torch/optim/sgd.py
"""

import torch
from torch.optim.optimizer import Optimizer, required

Expand Down

0 comments on commit 52c3769

Please sign in to comment.