diff --git a/logs/main_unit_test.log b/logs/main_unit_test.log
index a7a232f06da859ee49e317a53e7b790daa01c3d7..b7fd286ad67cba641e4a22b8dd7ed479a9763dd7 100644
--- a/logs/main_unit_test.log
+++ b/logs/main_unit_test.log
@@ -1451,3 +1451,15 @@ INFO:root:Generating the index
 INFO:root:Loading the index from sub_2CMEMS-MEDSEA-2017-testing.nc.bin_index.idx
 INFO:root:I loaded 112860 values in the test set
 INFO:root:= Filling in the submission file
+INFO:root:= Dataloaders for mean and standard deviation
+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 train fold has 542419 samples
diff --git a/main.py b/main.py
index dd2773a83d0802c5bcf2b4015b0a3baa3153c088..b511a718e4f609d02444ceacb5302e5470c094ae 100644
--- a/main.py
+++ b/main.py
@@ -40,7 +40,7 @@ if __name__ == "__main__":
     args = parser.parse_args()
 
     config_file = open("config.yml")
-    cfg = yaml.load(config_file)
+    cfg = yaml.load(config_file, Loader=yaml.FullLoader)
 
     rootDir = args.rootDir if args.rootDir != None else cfg["LogDir"]
 
diff --git a/sub_2CMEMS-MEDSEA-2010-2016-training.nc.bin_index.idx b/sub_2CMEMS-MEDSEA-2010-2016-training.nc.bin_index.idx
index eb294af77a8c600d154a1f18ccac54efa99a9b7d..0d347a032daa1ac0d652b9a7ed4a433d08a0f433 100644
Binary files a/sub_2CMEMS-MEDSEA-2010-2016-training.nc.bin_index.idx and b/sub_2CMEMS-MEDSEA-2010-2016-training.nc.bin_index.idx differ
diff --git a/train_indices.subset b/train_indices.subset
index 42c86043df55f960f18c6cf45c3704b40a70d0c6..5238f2b72ac26e10ef503aef30c8edad75a3c638 100644
Binary files a/train_indices.subset and b/train_indices.subset differ
diff --git a/valid_indices.subset b/valid_indices.subset
index bc5350bc44a87053c90c0530a149ff243d3f56e1..d7bfae7db48036e9d0573234d973dce5c55fab6c 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 da21843b9d82e5db6d4b9c142114a12b3aea9fd7..201421dc4c618735b49a29e0c8a9fdbe1d6c7240 120000
--- a/wandb/debug-internal.log
+++ b/wandb/debug-internal.log
@@ -1 +1 @@
-run-20230131_200825-ytr4zy6r/logs/debug-internal.log
\ No newline at end of file
+run-20230203_210417-nbeo06fq/logs/debug-internal.log
\ No newline at end of file
diff --git a/wandb/debug.log b/wandb/debug.log
index 749761fae83f97d3b8aeffeb8d5861b0a5a1a9ac..162bc4fc582d102c3d9c237d0daafb3422a4ee83 120000
--- a/wandb/debug.log
+++ b/wandb/debug.log
@@ -1 +1 @@
-run-20230131_200825-ytr4zy6r/logs/debug.log
\ No newline at end of file
+run-20230203_210417-nbeo06fq/logs/debug.log
\ No newline at end of file
diff --git a/wandb/latest-run b/wandb/latest-run
index d316e73edb34ef92b38536a3c73189a7a4c7f17f..dc1b371b55b39600231afca9ecea0843dd3c166b 120000
--- a/wandb/latest-run
+++ b/wandb/latest-run
@@ -1 +1 @@
-run-20230131_200825-ytr4zy6r
\ No newline at end of file
+run-20230203_210417-nbeo06fq
\ No newline at end of file