diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2022-08-08 11:44:28 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2022-08-08 11:44:28 +0800 |
| commit | 9245e96df5e4eab16b001817d0e3973b356e113f (patch) | |
| tree | 6797ac3dbebd8f7612edf0b5cd4f724b55b467c8 /simclr/main.py | |
| parent | 6920d0c033b18d4d27cdadce17f4b56347ed2a5a (diff) | |
Fix a typo
Diffstat (limited to 'simclr/main.py')
| -rw-r--r-- | simclr/main.py | 2 |
1 files changed, 1 insertions, 1 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, |