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