aboutsummaryrefslogtreecommitdiff
path: root/supervised
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2022-03-15 17:37:09 +0800
committerJordan Gong <jordan.gong@protonmail.com>2022-03-15 17:37:09 +0800
commit6942e46969e9d1476bf535e002356659121572dd (patch)
tree4d9e093aeefda7f7b8192a2a7bbfd68c94a1d5c6 /supervised
parentd7751c46315a6126950a283d8a32a8d14686ded2 (diff)
Change directories to project root
Diffstat (limited to 'supervised')
-rw-r--r--supervised/baseline.py6
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