From 9245e96df5e4eab16b001817d0e3973b356e113f Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Mon, 8 Aug 2022 11:44:28 +0800 Subject: Fix a typo --- simclr/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'simclr/main.py') 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, -- cgit v1.2.3