From bcaafc1eee9e9fe22684ef005fda74164a4a09af Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Tue, 9 Feb 2021 18:04:23 +0100
Subject: [PATCH] Final model configuration.
---
pysegcnn/main/train_config.py | 7 ++++---
1 file changed, 4 insertions(+), 3 deletions(-)
diff --git a/pysegcnn/main/train_config.py b/pysegcnn/main/train_config.py
index d83a6f2..fc54e23 100644
--- a/pysegcnn/main/train_config.py
+++ b/pysegcnn/main/train_config.py
@@ -31,7 +31,8 @@ HERE = pathlib.Path(__file__).resolve().parent
# path to the datasets on the current machine
# DRIVE_PATH = pathlib.Path('C:/Eurac/Projects/CCISNOW/Datasets/')
# DRIVE_PATH = pathlib.Path('/mnt/CEPH_PROJECTS/cci_snow/dfrisinghelli/Datasets/') # nopep8
-DRIVE_PATH = pathlib.Path('/home/dfrisinghelli/Datasets/')
+# DRIVE_PATH = pathlib.Path('/home/dfrisinghelli/Datasets/')
+DRIVE_PATH = pathlib.Path('/home/clusterusers/dfrisinghelli_eurac/Datasets/')
# name and paths to the datasets
DATASETS = {'Sparcs': DRIVE_PATH.joinpath('Sparcs'),
@@ -235,7 +236,7 @@ model_config = {
# define the batch size
# determines how many samples of the dataset are processed until the
# weights of the network are updated (via mini-batch gradient descent)
- 'batch_size': 128,
+ 'batch_size': 256,
# the seed for the random number generator intializing the network weights
'torch_seed': 0,
@@ -254,6 +255,6 @@ model_config = {
# define the number of epochs: the number of maximum iterations over
# the whole training dataset
- 'epochs': 5000,
+ 'epochs': 200,
}
--
GitLab