From 8688e3eb439b09893cfb43901367c5cb4b3c23bd Mon Sep 17 00:00:00 2001 From: Natalia Gimelshein Date: Tue, 19 Jul 2016 16:55:06 -0700 Subject: saving states works set tensors to nil when resetting states don't silently nil hidden states, add rememberStates to GRU --- LSTM.lua | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'LSTM.lua') 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 -- cgit v1.2.3