diff options
author | Soumith Chintala <soumith@gmail.com> | 2016-03-22 18:45:17 +0300 |
---|---|---|
committer | Soumith Chintala <soumith@gmail.com> | 2016-03-22 18:45:17 +0300 |
commit | 7e74cb7464106114b468289f5dcc59744b2d25cd (patch) | |
tree | 1b81e6d171626bf68ceb5afdd6dcce18dbbb1a96 | |
parent | 864aba149f3b7102c1e02481d801d289fffa6238 (diff) | |
parent | 69c0f1a6281cd01fc201779674e1df63afb219d8 (diff) |
Merge pull request #148 from szagoruyko/conv-pad-check
Check if padW and padH exist in SpatialConvolution
-rw-r--r-- | SpatialConvolution.lua | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/SpatialConvolution.lua b/SpatialConvolution.lua index 3be7715..9a78711 100644 --- a/SpatialConvolution.lua +++ b/SpatialConvolution.lua @@ -120,6 +120,7 @@ function SpatialConvolution:createIODescriptors(input) -- create conv descriptor self.convDesc = ffi.new('struct cudnnConvolutionStruct*[1]') errcheck('cudnnCreateConvolutionDescriptor', self.convDesc) + self.padH, self.padW = self.padH or 0, self.padW or 0 local pad = torch.IntTensor({self.padH, self.padW}) local stride = torch.IntTensor({self.dH, self.dW}) local upscale = torch.IntTensor({1,1}) |