diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2022-03-15 17:59:24 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2022-03-15 17:59:24 +0800 |
| commit | 587ccee452d18c44cf33523350b16c79c485d0e7 (patch) | |
| tree | e9cb1620f3d8694f1ea3e0f59b1ebadbbf7d5d9e /supervised/baseline.py | |
| parent | 6942e46969e9d1476bf535e002356659121572dd (diff) | |
Fix module path
Diffstat (limited to 'supervised/baseline.py')
| -rw-r--r-- | supervised/baseline.py | 4 |
1 files 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' |