-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.py
21 lines (19 loc) · 990 Bytes
/
config.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
# -----------------
# DATASET ROOTS
# -----------------
cifar_10_root = '/data4/datasets/CIFAR/'
cifar_100_root = '/data4/datasets/CIFAR/'
tiny_imagenet_root = '/data4/datasets/TinyImageNet/'
cub_root = '/data4/datasets/gcd/CUB/'
aircraft_root = '/data4/datasets/gcd/FGVC-Aircraft/fgvc-aircraft-2013b/'
car_root = "/data4/datasets/gcd/Stanford-Cars/"
herbarium_dataroot = '/data4/datasets/gcd/Herbarium19-Small/'
imagenet_root = '/data4/sjma/dataset/ImageNet/ILSVRC2012/imagenet/'
# OSR Split dir
osr_split_dir = '/data4/datasets/gcd/ssb_splits/'
# -----------------
# OTHER PATHS
# -----------------
dino_pretrain_path = '/data6/home/sjma/.cache/torch/hub/checkpoints/dino_vitbase16_pretrain.pth'
feature_extract_dir = '/lustre/home/sjma/Continual-GCD-project/GCD-baseline/extracted_features_public_impl' # Extract features to this directory
exp_root_happy = '/lustre/home/sjma/Continual-GCD-project/MyCGCD-v3/dev_outputs_Happy' # All logs and checkpoints will be saved here