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