From 9245e96df5e4eab16b001817d0e3973b356e113f Mon Sep 17 00:00:00 2001
From: Jordan Gong <jordan.gong@protonmail.com>
Date: Mon, 8 Aug 2022 11:44:28 +0800
Subject: Fix a typo
---
simclr/main.py | 2 +-
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,
--
cgit v1.2.3