Added missing changes

1 files changed, 2 insertions, 2 deletions

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