diff --git a/logs/main_unit_test.log b/logs/main_unit_test.log
index f9c0a53ef7ff80542a153d83a7eef4fb9be74ffd..1f73c40c840b15e9b847b805412d2ff4f3c45812 100644
--- a/logs/main_unit_test.log
+++ b/logs/main_unit_test.log
@@ -2452,3 +2452,33 @@ 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 541214 samples
 INFO:root:  - The valid fold has 135946 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 541896 samples
+INFO:root:  - The valid fold has 135264 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 541474 samples
+INFO:root:  - The valid fold has 135686 samples
diff --git a/model.py b/model.py
index e0357c26028787515848e2d5fe8540c527da772a..a16c567438e792017bceb1bb9250c6fb438acf41 100644
--- a/model.py
+++ b/model.py
@@ -77,7 +77,7 @@ class BidirectionalLSTM(nn.Module):
         self.hidden_size = cfg["BidirectionalLSTM"]["HiddenSize"]
         self.num_layers =  cfg["BidirectionalLSTM"]["NumLayers"]
         self.LSTM_dropout    =  cfg["BidirectionalLSTM"]["LSTMDropout"]
-        self.FFN_dropout    = cfg["DidirectionalLSTM"]["FFNDropout"]
+        self.FFN_dropout    = cfg["BidirectionalLSTM"]["FFNDropout"]
         self.num_ffn    = cfg["BidirectionalLSTM"]["NumFFN"]
 
         self.lstm = nn.LSTM(input_size, self.hidden_size, self.num_layers, batch_first = True, bidirectional =True, dropout = self.LSTM_dropout)
diff --git a/train_indices.subset b/train_indices.subset
index 4be28c3fddee04d316099595035195d190335a28..594eede69b0c69b6daf7fbbf6ce1e45ddce43a8d 100644
Binary files a/train_indices.subset and b/train_indices.subset differ
diff --git a/valid_indices.subset b/valid_indices.subset
index 71e2f411ff24dec16c1ec6873ea24d6a85512515..8ca27a17829f8e1391a93beb780eacbc9e41219b 100644
Binary files a/valid_indices.subset and b/valid_indices.subset differ