diff --git a/job.py b/job.py
index 6349a74a86d671aa9d7a3003d823ecb126a92717..10c9d7d7f1aed63098cc95ca40705baefcef081c 100644
--- a/job.py
+++ b/job.py
@@ -78,6 +78,10 @@ parser.add_argument("--model_name",
                     default ="Bi-LSTM",
                     help="Name of the model to train")
 
+parser.add_argument("--experimentName",
+                    default ="",
+                    help="Name of the current experiment")
+
 # Ensure the log directory exists
 os.system("mkdir -p logslurms")
 
diff --git a/logs/main_unit_test.log b/logs/main_unit_test.log
index f72c143f1f4ad14af28205025b9ef3478c3c4567..6ddb9bffc3ac8cd6903aecafdeabc3a840416940 100644
--- a/logs/main_unit_test.log
+++ b/logs/main_unit_test.log
@@ -2721,3 +2721,18 @@ INFO:root:The loaded dataset contains 25 latitudes, 37 longitudes, 28 depths and
 INFO:root:Loading the index from sub_2CMEMS-MEDSEA-2010-2016-training.nc.bin_index.idx
 INFO:root:  - The train fold has 541609 samples
 INFO:root:  - The valid fold has 135551 samples
+INFO:root:= Dataloaders
+INFO:root:  - Dataset creation
+INFO:root:The loaded dataset contains 25 latitudes, 37 longitudes, 28 depths and 2222 time points
+INFO:root:Generating the index
+INFO:root:Loading the index from sub_2CMEMS-MEDSEA-2010-2016-training.nc.bin_index.idx
+INFO:root:  - Loaded a dataset with 677160 samples
+INFO:root:  - Splitting the data in training and validation sets
+INFO:root:Generating the subset files from 677160 samples
+INFO:root:  - Subset dataset
+INFO:root:The loaded dataset contains 25 latitudes, 37 longitudes, 28 depths and 2222 time points
+INFO:root:Loading the index from sub_2CMEMS-MEDSEA-2010-2016-training.nc.bin_index.idx
+INFO:root:The loaded dataset contains 25 latitudes, 37 longitudes, 28 depths and 2222 time points
+INFO:root:Loading the index from sub_2CMEMS-MEDSEA-2010-2016-training.nc.bin_index.idx
+INFO:root:  - The train fold has 541621 samples
+INFO:root:  - The valid fold has 135539 samples
diff --git a/train_indices.subset b/train_indices.subset
index 801ef942d91e6e4dde955d1555b6be9e784a8e5f..775627b199394f7334d13628767253fc70ee8f4a 100644
Binary files a/train_indices.subset and b/train_indices.subset differ
diff --git a/valid_indices.subset b/valid_indices.subset
index 16b6fc468d117d472925c9975058cda1453fb00c..5dade12f6242af965e01d887b87463504aea3f78 100644
Binary files a/valid_indices.subset and b/valid_indices.subset differ
diff --git a/wandb/debug-internal.log b/wandb/debug-internal.log
index 86e3c6db85a338ab8d678250d88aa44012db1885..7b6c88ad31abd2aa0412a30e76943f006c595653 120000
--- a/wandb/debug-internal.log
+++ b/wandb/debug-internal.log
@@ -1 +1 @@
-run-20230204_131220-udsbqzli/logs/debug-internal.log
\ No newline at end of file
+run-20230211_212625-o8jq0eya/logs/debug-internal.log
\ No newline at end of file
diff --git a/wandb/debug.log b/wandb/debug.log
index f159925d26b61b22c8fb7b0fc268b4daba398c32..14b2e91e2bef1da8a8ae3da25fb60f268d7a151c 120000
--- a/wandb/debug.log
+++ b/wandb/debug.log
@@ -1 +1 @@
-run-20230204_131220-udsbqzli/logs/debug.log
\ No newline at end of file
+run-20230211_212625-o8jq0eya/logs/debug.log
\ No newline at end of file
diff --git a/wandb/latest-run b/wandb/latest-run
index c44486eefedb3139505e9e4dd55a4a1c8354a55b..921df60f4e49ce33f24f5b4449ce8089a4288094 120000
--- a/wandb/latest-run
+++ b/wandb/latest-run
@@ -1 +1 @@
-run-20230204_131220-udsbqzli
\ No newline at end of file
+run-20230211_212625-o8jq0eya
\ No newline at end of file