From d91a5ef629a3432ad38072221ae2dc8c5b90f576 Mon Sep 17 00:00:00 2001
From: Yandi <yandirzm@gmail.com>
Date: Sat, 11 Feb 2023 22:26:42 +0100
Subject: [PATCH] bilstm with dropout

---
 config.yml              |  2 +-
 logs/main_unit_test.log | 15 +++++++++++++++
 2 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/config.yml b/config.yml
index ca6f741..6c62c55 100644
--- a/config.yml
+++ b/config.yml
@@ -64,7 +64,7 @@ LinearRegression:
 BidirectionalLSTM:
   HiddenSize: 16
   NumLayers: 2
-  LSTMDropout: 0
+  LSTMDropout: 0.2
   FFNDropout: 0
   NumFFN: 3
   Initialization: init_he
diff --git a/logs/main_unit_test.log b/logs/main_unit_test.log
index e7d7b07..69191eb 100644
--- a/logs/main_unit_test.log
+++ b/logs/main_unit_test.log
@@ -2766,3 +2766,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 542221 samples
 INFO:root:  - The valid fold has 134939 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 542036 samples
+INFO:root:  - The valid fold has 135124 samples
-- 
GitLab