diff options
| -rw-r--r-- | simclr/main.py | 2 | ||||
| -rw-r--r-- | supervised/baseline.py | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/simclr/main.py b/simclr/main.py index 9701a02..4dfdcf9 100644 --- a/simclr/main.py +++ b/simclr/main.py @@ -242,7 +242,7 @@ class SimCLRTrainer(Trainer): lr=optim_config.lr, betas=optim_config.betas, ) - elif optim_config.optim in {'sdg', 'lars'}: + elif optim_config.optim in {'sgd', 'lars'}: optimizer = torch.optim.SGD( param_groups, lr=optim_config.lr, diff --git a/supervised/baseline.py b/supervised/baseline.py index a9d9862..9ad107c 100644 --- a/supervised/baseline.py +++ b/supervised/baseline.py @@ -190,7 +190,7 @@ class SupBaselineTrainer(Trainer): lr=optim_config.lr, betas=optim_config.betas, ) - elif optim_config.optim == 'sdg': + elif optim_config.optim == 'sgd': optimizer = torch.optim.SGD( param_groups, lr=optim_config.lr, |