Spaces:
Runtime error
Runtime error
File size: 348 Bytes
982b011 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 | WORK_DIR = "./"
DATA_DIR = f"{WORK_DIR}/data"
IMAGES_DIR = f"{DATA_DIR}/images"
RESULTS_DIR = f"{WORK_DIR}/results"
# supported feature extractor models
FEATURE_EXTRACTOR_MODELS = [
"resnet18",
"resnet34",
"resnet50",
"resnet101",
"resnet152",
"vit_b_16",
"vit_b_32",
"vit_l_16",
"vit_l_32",
"vit_h_14",
]
|