From 587ccee452d18c44cf33523350b16c79c485d0e7 Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Tue, 15 Mar 2022 17:59:24 +0800 Subject: Fix module path --- supervised/baseline.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'supervised/baseline.py') 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