diff options
| author | Martin Simonovsky <simonovm@imagine.enpc.fr> | 2015-09-25 13:04:05 +0300 |
|---|---|---|
| committer | Martin Simonovsky <simonovm@imagine.enpc.fr> | 2015-10-02 21:05:08 +0300 |
| commit | 5e3560b41367809981811f7328417f369b36beeb (patch) | |
| tree | 683cf6e1c5ffefe55ba1e8d773c40a613b11502a | |
| parent | c583b7502316a62705b6a355afcd4dcf556f96f2 (diff) | |
Change default value of padH to be consistent with nn convolution modules (=padW)
| -rw-r--r-- | README.md | 2 | ||||
| -rw-r--r-- | SpatialConvolution.lua | 4 |
2 files changed, 2 insertions, 4 deletions
@@ -15,7 +15,7 @@ Modules are API compatible their [`nn`](https://github.com/torch/nn) equivalents ```lua -- All inputs have to be 3D or 4D(batch-mode), except ReLU, Tanh and Sigmoid -cudnn.SpatialConvolution(nInputPlane, nOutputPlane, kW, kH, [dW = 1], [dH = 1], [padW = 0], [padH = 0], [groups = 1]) +cudnn.SpatialConvolution(nInputPlane, nOutputPlane, kW, kH, [dW = 1], [dH = 1], [padW = 0], [padH = padW], [groups = 1]) cudnn.SpatialMaxPooling(kW, kH, dW, dH, padW, padH) cudnn.SpatialAveragePooling(kW, kH, dW, dH, padW, padH) diff --git a/SpatialConvolution.lua b/SpatialConvolution.lua index c67130a..9059801 100644 --- a/SpatialConvolution.lua +++ b/SpatialConvolution.lua @@ -12,10 +12,8 @@ function SpatialConvolution:__init(nInputPlane, nOutputPlane, kW, kH, dW, dH, padW, padH, groups) local delayedReset = self.reset self.reset = function() end - parent.__init(self, nInputPlane, nOutputPlane, kW, kH, dW, dH) + parent.__init(self, nInputPlane, nOutputPlane, kW, kH, dW, dH, padW, padH) self.reset = delayedReset - self.padW = padW or 0 - self.padH = padH or 0 self.groups = groups or 1 assert(nInputPlane % self.groups == 0, 'nInputPlane should be divisible by nGroups') |