Add BiLSTM (microsoft#492)

Irisated · Apr 15, 2020 · 3953cca · 3953cca
1 parent b4f6e13
commit 3953cca
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/A-基础教程/A2-神经网络基本原理简明教程/SourceCode/ch19-RNNBasic/Level8_Pytorch_LSTM.py b/A-基础教程/A2-神经网络基本原理简明教程/SourceCode/ch19-RNNBasic/Level8_Pytorch_LSTM.py
@@ -48,9 +48,10 @@ def __init__(self):
         super(RNN, self).__init__()
         self.rnn = nn.LSTM(
             input_size=29,          # character num.
-            hidden_size=16,         # RNN or LSTM hidden layer, 设置的稍大一些可能效果更佳，此处仅作对比
+            hidden_size=8,         # RNN or LSTM hidden layer, 设置的稍大一些可能效果更佳，此处仅作对比
             num_layers=1,
             batch_first=True,
+            bidirectional=True      # 双向LSTM, 若设置为False,对应hidden_size增大两倍
         )
         self.out = nn.Linear(16, 10)
         self.softmax = nn.Softmax()         # classification, softmax