Update function name for __main__ in ON_LSTM

yikangshen · May 17, 2019 · 975a8f3 · 975a8f3
1 parent 8307753
commit 975a8f3
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])