diff options
Diffstat (limited to 'LSTM.lua')
-rw-r--r-- | LSTM.lua | 4 |
1 files changed, 2 insertions, 2 deletions
@@ -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 |