diff options
| -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' |