andras/bipolaroidbipolaroid
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Extract paths

Browse source
This commit is contained in:
Andras Schmelczer 2024-06-16 20:50:51 +01:00
parent e6c0b3d2d9
commit 615dafccc8
No known key found for this signature in database
GPG key ID: FC8F2C3D3D1A718C
1 changed files with 12 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

16
src/config.py
View file

@ -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)