From 587ccee452d18c44cf33523350b16c79c485d0e7 Mon Sep 17 00:00:00 2001
From: Jordan Gong <jordan.gong@protonmail.com>
Date: Tue, 15 Mar 2022 17:59:24 +0800
Subject: Fix module path

---
 supervised/baseline.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/supervised/baseline.py b/supervised/baseline.py
index 99f3fa1..6a83f72 100644
--- a/supervised/baseline.py
+++ b/supervised/baseline.py
@@ -12,8 +12,8 @@ from torchvision.models.resnet import BasicBlock
 from torchvision.transforms import transforms, InterpolationMode
 from tqdm import tqdm
 
-from supervised.lars_optimizer import LARS
-from supervised.scheduler import LinearWarmupAndCosineAnneal
+from lars_optimizer import LARS
+from scheduler import LinearWarmupAndCosineAnneal
 
 CODENAME = 'cifar10-resnet50-aug-lars-sched'
 DATASET_ROOT = 'dataset'
-- 
cgit v1.2.3