From 615dafccc811163b97bf05ebb9ec00cceb72e5f6 Mon Sep 17 00:00:00 2001 From: Andras Schmelczer Date: Sun, 16 Jun 2024 20:50:51 +0100 Subject: [PATCH] Extract paths --- src/config.py | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) 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)