diff --git a/src/config.py b/src/config.py index 9e25b17..1e0eaa2 100644 --- a/src/config.py +++ b/src/config.py @@ -3,8 +3,16 @@ from pathlib import Path DATA = sorted(Path("/mnt/wsl/PHYSICALDRIVE1/data/unsplash").glob("*.jpg")) TRAIN_SIZE = 0.9 -CACHE_PATH = Path("/mnt/wsl/PHYSICALDRIVE1/data/cache") -CACHE_PATH.mkdir(exist_ok=True, parents=True) - +CACHE_PATH = Path("/mnt/wsl/PHYSICALDRIVE1/data/cache2") MODELS_PATH = Path("/home/andras/projects/bipolaroid/models") -MODELS_PATH.mkdir(exist_ok=True, parents=True) +LOGS_PATH = Path("/home/andras/projects/bipolaroid/logs") +RUNS_PATH = Path("/home/andras/projects/bipolaroid/runs") + + +for path in [ + CACHE_PATH, + MODELS_PATH, + LOGS_PATH, + RUNS_PATH +]: + path.mkdir(exist_ok=True, parents=True)