From d716d9f265d19cf529d9e62e702e1409a28018f5 Mon Sep 17 00:00:00 2001 From: Joan Puigcerver Date: Tue, 5 Jul 2016 11:27:19 +0200 Subject: Added missing changes --- LSTM.lua | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'LSTM.lua') diff --git a/LSTM.lua b/LSTM.lua index 29c199c..d9f10a2 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) - parent.__init(self,inputSize, hiddenSize, numLayers, batchFirst) +function LSTM:__init(inputSize, hiddenSize, numLayers, batchFirst, dropout) + parent.__init(self,inputSize, hiddenSize, numLayers, batchFirst, dropout) self.mode = 'CUDNN_LSTM' self:reset() end -- cgit v1.2.3