Merge pull request #7 from ArkAung/master

Update function name in ON_LSTM.py
yikangshen · May 21, 2019 · 46d63cd · 46d63cd
2 parents 8307753 + 975a8f3
commit 46d63cd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ON_LSTM.py b/ON_LSTM.py
@@ -179,6 +179,6 @@ def forward(self, input, hidden):
 if __name__ == "__main__":
     x = torch.Tensor(10, 10, 10)
     x.data.normal_()
-    lstm = LSTMCellStack([10, 10, 10])
+    lstm = ONLSTMStack([10, 10, 10], chunk_size=10)
     print(lstm(x, lstm.init_hidden(10))[1])