Skip to content
Snippets Groups Projects
Commit f6f4175f authored by Yandi's avatar Yandi
Browse files

bilstm trying to deal with NaN by creating a shallower network

parent 4d3f8c89
No related branches found
No related tags found
1 merge request!1Master into main
Hide whitespace changes
Inline Side-by-side
config.yml
+ 4
− 4
View file @ f6f4175f
...@@ -58,12 +58,12 @@ LinearRegression: ...@@ -58,12 +58,12 @@ LinearRegression:
Initialization: init_he Initialization: init_he
BidirectionalLSTM: BidirectionalLSTM:
HiddenSize: 32 HiddenSize: 16
NumLayers: 4 NumLayers: 2
LSTMDropout: 0 LSTMDropout: 0
FFNDropout: 0.2 FFNDropout: 0.2
NumFFN: 8 NumFFN: 3
Initialization: None Initialization: init_he
RNN: RNN:
HiddenSize: 32 HiddenSize: 32
......
logs/main_unit_test.log
+ 15
− 0
View file @ f6f4175f
No preview for this file type
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment