diff --git a/logs/main_unit_test.log b/logs/main_unit_test.log
index 5c0f5e91fb2961544620e36339b9345dbec38de9..be53f85da17e17425bcb7209cf69fd365772c0fb 100644
--- a/logs/main_unit_test.log
+++ b/logs/main_unit_test.log
@@ -2497,3 +2497,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 542034 samples
 INFO:root:  - The valid fold has 135126 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 542427 samples
+INFO:root:  - The valid fold has 134733 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 541893 samples
+INFO:root:  - The valid fold has 135267 samples
diff --git a/main.py b/main.py
index f21ba8b0697409ad2189813c71262b385a471518..8e568cd3b2276852ab45e8db4c4038c0ea44b1fc 100644
--- a/main.py
+++ b/main.py
@@ -121,6 +121,8 @@ if __name__ == "__main__":
         wandb.run.name = raw_run_name
         wandb.watch(network, log_freq = log_freq)
 
+    torch.autograd.set_detect_anomaly(True)
+    
     for t in range(cfg["Training"]["Epochs"]):
         print("Epoch {}".format(t))
         train(args, network, train_loader, f_loss, optimizer, device, log_interval)
diff --git a/train_indices.subset b/train_indices.subset
index 594eede69b0c69b6daf7fbbf6ce1e45ddce43a8d..6b03f35655aa0ac4dd73d9dbe8d7962d868b9e0a 100644
Binary files a/train_indices.subset and b/train_indices.subset differ
diff --git a/valid_indices.subset b/valid_indices.subset
index 8ca27a17829f8e1391a93beb780eacbc9e41219b..b6bc2b9147d2278d5d3897fa11f4e85722d72e6c 100644
Binary files a/valid_indices.subset and b/valid_indices.subset differ