diff --git a/config.yml b/config.yml
index 22a089affe4b11f729c9ae22fe2d3b09e9df803f..7ed0a0c410ea6afa5e199e2ca4b67c1af02c0f07 100644
--- a/config.yml
+++ b/config.yml
@@ -56,8 +56,8 @@ BidirectionalLSTM:
 
 RNN:
   HiddenSize: 35
-  NumLayers: 4
-  NumFFN: 15
+  NumLayers: 2
+  NumFFN: 2
   Dropout: 0.2
   Initialization: None
 
diff --git a/logs/main_unit_test.log b/logs/main_unit_test.log
index 6feec89f290557e58d7a58b53b59f22c91620b18..18b8c71acf0f308d3979643fbb155ea1635e90c8 100644
--- a/logs/main_unit_test.log
+++ b/logs/main_unit_test.log
@@ -1622,3 +1622,111 @@ 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 541670 samples
 INFO:root:  - The valid fold has 135490 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 542299 samples
+INFO:root:  - The valid fold has 134861 samples
+INFO:root:Epoch 0
+INFO:root: Validation : Loss : nan
+INFO:root:Epoch 1
+INFO:root: Validation : Loss : nan
+INFO:root:Epoch 2
+INFO:root: Validation : Loss : nan
+INFO:root:Epoch 3
+INFO:root: Validation : Loss : nan
+INFO:root:Epoch 4
+INFO:root: Validation : Loss : nan
+INFO:root:Epoch 5
+INFO:root: Validation : Loss : nan
+INFO:root:Epoch 6
+��������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������INFO:root: Validation : Loss : nan
+INFO:root:Epoch 7
+INFO:root: Validation : Loss : nan
+INFO:root:Epoch 8
+INFO:root: Validation : Loss : nan
+INFO:root:Epoch 9
+INFO:root: Validation : Loss : nan
+INFO:root:Epoch 10
+INFO:root: Validation : Loss : nan
+INFO:root:Epoch 11
+INFO:root: Validation : Loss : nan
+INFO:root:Epoch 12
+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 541854 samples
+INFO:root:  - The valid fold has 135306 samples
+INFO:root: Validation : Loss : nan
+INFO:root:Epoch 13
+INFO:root: Validation : Loss : nan
+INFO:root:Epoch 14
+INFO:root: Validation : Loss : nan
+INFO:root:Epoch 15
+INFO:root: Validation : Loss : nan
+INFO:root:Epoch 16
+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 541663 samples
+INFO:root:  - The valid fold has 135497 samples
+INFO:root: Validation : Loss : nan
+INFO:root:Epoch 17
+INFO:root: Validation : Loss : nan
+INFO:root:Epoch 18
+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
+INFO:root: Validation : Loss : nan
+INFO:root:Epoch 19
+INFO:root: Validation : Loss : nan
+INFO:root:Building the dataloader
+INFO:root:= Dataloaders
+INFO:root:  - Dataset creation
+INFO:root:The loaded dataset contains 25 latitudes, 37 longitudes, 28 depths and 365 time points
+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
diff --git a/main.py b/main.py
index e1089f91e017886fdd5ff0bbf2818037e24597d4..e5edb4ad7abe53794cf3328b6ffe084b6f5afa23 100644
--- a/main.py
+++ b/main.py
@@ -120,14 +120,14 @@ if __name__ == "__main__":
         wandb.watch(network, log_freq = log_freq)
 
     for t in range(cfg["Training"]["Epochs"]):
-        logging.info("Epoch {}".format(t))
+        print("Epoch {}".format(t))
         train(args, network, train_loader, f_loss, optimizer, device, log_interval)
 
         val_loss = test.test(network, valid_loader, f_loss, device)
 
         network_checkpoint.update(val_loss)
 
-        logging.info(" Validation : Loss : {:.4f}".format(val_loss))
+        print(" Validation : Loss : {:.4f}".format(val_loss))
         if not args.no_wandb:
             wandb.log({"val_loss": val_loss})
 
diff --git a/test.py b/test.py
index 2007732fc856ce8cee1f3868ecb19e759b182387..b2142768dc581b4ef91ec41d064de49bbf19485f 100644
--- a/test.py
+++ b/test.py
@@ -35,6 +35,10 @@ def test(model, loader, f_loss, device):
 
             # Compute the forward pass, i.e. the scores for each input image
             outputs = model(inputs)
+            print("Validation inputs :")
+            print(inputs)
+            print("Validation outputs :")
+            print(outputs)
 
             # We accumulate the exact number of processed samples
             N += inputs.shape[0]
diff --git a/train_indices.subset b/train_indices.subset
index d5948c507d73de00e7c948102ebfea539209a40d..908f94d84b96fa83e570be8e0d757dc2a204c211 100644
Binary files a/train_indices.subset and b/train_indices.subset differ
diff --git a/valid_indices.subset b/valid_indices.subset
index d0755c597744bcb37c9eca6a9d1e71430d05ba7c..db33627debbe3962e9335062ed98d346d4ab0f11 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 6a852e786fa2c05e19a98111fd7d9bc3c00639ee..7405266d86a1dca74b8f6b1a13e5172fbea0a14e 120000
--- a/wandb/debug-internal.log
+++ b/wandb/debug-internal.log
@@ -1 +1 @@
-run-20230203_215737-zxchwf43/logs/debug-internal.log
\ No newline at end of file
+run-20230203_233455-ujner2cx/logs/debug-internal.log
\ No newline at end of file
diff --git a/wandb/debug.log b/wandb/debug.log
index f441fad8d1a61f2b965b9d8f8f55793517280689..d2a2374152daf8a98632afd89acf22f36b0e9e4c 120000
--- a/wandb/debug.log
+++ b/wandb/debug.log
@@ -1 +1 @@
-run-20230203_215737-zxchwf43/logs/debug.log
\ No newline at end of file
+run-20230203_233455-ujner2cx/logs/debug.log
\ No newline at end of file
diff --git a/wandb/latest-run b/wandb/latest-run
index f5aacd3beb2c843c21f3e9dd475ca9d49263a27a..c0eef7e6f30408ede8d6007a30e95ab9b7701a24 120000
--- a/wandb/latest-run
+++ b/wandb/latest-run
@@ -1 +1 @@
-run-20230203_215737-zxchwf43
\ No newline at end of file
+run-20230203_233455-ujner2cx
\ No newline at end of file