saving states works

set tensors to nil when resetting states

don't silently nil hidden states, add rememberStates to GRU

1 files changed, 2 insertions, 2 deletions

diff --git a/LSTM.lua b/LSTM.lua
index d9f10a2..69b3322 100644
--- a/LSTM.lua
+++ b/LSTM.lua
@@ -1,7 +1,7 @@
 local LSTM, parent = torch.class('cudnn.LSTM', 'cudnn.RNN')
 
-function LSTM:__init(inputSize, hiddenSize, numLayers, batchFirst, dropout)
-    parent.__init(self,inputSize, hiddenSize, numLayers, batchFirst, dropout)
+function LSTM:__init(inputSize, hiddenSize, numLayers, batchFirst, dropout, rememberStates)
+    parent.__init(self,inputSize, hiddenSize, numLayers, batchFirst, dropout, rememberStates)
     self.mode = 'CUDNN_LSTM'
     self:reset()
 end