diff options
Diffstat (limited to 'supervised/baseline.py')
| -rw-r--r-- | supervised/baseline.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/supervised/baseline.py b/supervised/baseline.py index 047d8ca..99f3fa1 100644 --- a/supervised/baseline.py +++ b/supervised/baseline.py @@ -16,9 +16,9 @@ from supervised.lars_optimizer import LARS from supervised.scheduler import LinearWarmupAndCosineAnneal CODENAME = 'cifar10-resnet50-aug-lars-sched' -DATASET_ROOT = '../dataset' -TENSORBOARD_PATH = os.path.join('../runs', CODENAME) -CHECKPOINT_PATH = os.path.join('../checkpoints', CODENAME) +DATASET_ROOT = 'dataset' +TENSORBOARD_PATH = os.path.join('runs', CODENAME) +CHECKPOINT_PATH = os.path.join('checkpoints', CODENAME) BATCH_SIZE = 256 N_EPOCHS = 1000 |