diff --git a/ConvLSTM.lua b/ConvLSTM.lua index c356a49..bcb9bfa 100644 --- a/ConvLSTM.lua +++ b/ConvLSTM.lua @@ -15,6 +15,8 @@ require 'extracunn' local ConvLSTM, parent = torch.class('nn.ConvLSTM', 'nn.LSTM') function ConvLSTM:__init(inputSize, outputSize, rho, kc, km, stride, batchSize) + assert(km % 2 == 1, 'ConvLSTM requires that km MUST be an odd number') + self.kc = kc self.km = km self.padc = torch.floor(kc/2)