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

fixing typo d instead of b

parent 2eb29787
No related branches found
No related tags found
1 merge request!1Master into main
No preview for this file type
...@@ -77,7 +77,7 @@ class BidirectionalLSTM(nn.Module): ...@@ -77,7 +77,7 @@ class BidirectionalLSTM(nn.Module):
self.hidden_size = cfg["BidirectionalLSTM"]["HiddenSize"] self.hidden_size = cfg["BidirectionalLSTM"]["HiddenSize"]
self.num_layers = cfg["BidirectionalLSTM"]["NumLayers"] self.num_layers = cfg["BidirectionalLSTM"]["NumLayers"]
self.LSTM_dropout = cfg["BidirectionalLSTM"]["LSTMDropout"] self.LSTM_dropout = cfg["BidirectionalLSTM"]["LSTMDropout"]
self.FFN_dropout = cfg["DidirectionalLSTM"]["FFNDropout"] self.FFN_dropout = cfg["BidirectionalLSTM"]["FFNDropout"]
self.num_ffn = cfg["BidirectionalLSTM"]["NumFFN"] self.num_ffn = cfg["BidirectionalLSTM"]["NumFFN"]
self.lstm = nn.LSTM(input_size, self.hidden_size, self.num_layers, batch_first = True, bidirectional =True, dropout = self.LSTM_dropout) self.lstm = nn.LSTM(input_size, self.hidden_size, self.num_layers, batch_first = True, bidirectional =True, dropout = self.LSTM_dropout)
......
No preview for this file type
No preview for this file type
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment