diff options
author | Clement Farabet <clement.farabet@gmail.com> | 2012-05-04 22:05:25 +0400 |
---|---|---|
committer | Clement Farabet <clement.farabet@gmail.com> | 2012-05-04 22:05:25 +0400 |
commit | e35c60df0e711f3d4a053f95bcf2d7f1d59a708f (patch) | |
tree | 3dbf97d7227566e393cb0bad63d6b677088896e5 /SpatialPadding.lua | |
parent | 300bd31b0ec5bc108ce78b4f3730f8c67fc5a8bc (diff) |
Added defaults for backward compat.
Diffstat (limited to 'SpatialPadding.lua')
-rw-r--r-- | SpatialPadding.lua | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/SpatialPadding.lua b/SpatialPadding.lua index 91b172e..a61e4af 100644 --- a/SpatialPadding.lua +++ b/SpatialPadding.lua @@ -24,6 +24,8 @@ function SpatialPadding:__init(pad_l, pad_r, pad_t, pad_b, y_dim, x_dim) end function SpatialPadding:updateOutput(input) + self.x_dim = x_dim or 3 + self.y_dim = y_dim or 2 local h = input:size(self.y_dim) + self.pad_t + self.pad_b local w = input:size(self.x_dim) + self.pad_l + self.pad_r if w < 1 or h < 1 then error('input is too small') end |