From 97e8cb1fe14aac994a1411b95a6446e0b2c77073 Mon Sep 17 00:00:00 2001 From: Andras Schmelczer Date: Tue, 25 Jun 2024 08:25:20 +0100 Subject: [PATCH] Change models path --- .gitignore | 5 ++--- src/config.py | 2 +- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/.gitignore b/.gitignore index 9ef6031..585abc0 100644 --- a/.gitignore +++ b/.gitignore @@ -1,5 +1,4 @@ __pycache__ -runs +runs* *.log -models/*.pth -models/*.json +saved_models/* diff --git a/src/config.py b/src/config.py index 8651565..2a26230 100644 --- a/src/config.py +++ b/src/config.py @@ -5,7 +5,7 @@ DATA = sorted(Path("/mnt/wsl/PHYSICALDRIVE1/data/unsplash").glob("*.jpg")) TRAIN_SIZE = 0.9 CACHE_PATH = Path("/mnt/wsl/PHYSICALDRIVE1/data/cache2") -MODELS_PATH = Path("/home/andras/projects/bipolaroid/models") +MODELS_PATH = Path("/home/andras/projects/bipolaroid/saved_models") LOGS_PATH = Path("/home/andras/projects/bipolaroid/logs") RUNS_PATH = Path("/home/andras/projects/bipolaroid/runs")