diff options
| author | Soumith Chintala <soumith@gmail.com> | 2014-12-19 02:36:12 +0300 |
|---|---|---|
| committer | Soumith Chintala <soumith@gmail.com> | 2014-12-19 23:28:13 +0300 |
| commit | ad958c0e268d876ee4d713510b8c3ef83b37bca0 (patch) | |
| tree | 0defbe1196f778c9fb3f79f5f6e7a1da9ae92cda /SpatialMaxPooling.lua | |
| parent | d290c4cb9d632120d3fba97caefb3afb961081bf (diff) | |
everything works with R2. all unit tests pass. Maxpooling has free zero-padding
Diffstat (limited to 'SpatialMaxPooling.lua')
| -rw-r--r-- | SpatialMaxPooling.lua | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/SpatialMaxPooling.lua b/SpatialMaxPooling.lua index 108a055..e9a7b89 100644 --- a/SpatialMaxPooling.lua +++ b/SpatialMaxPooling.lua @@ -1,6 +1,6 @@ local SpatialMaxPooling, parent = torch.class('cudnn.SpatialMaxPooling', 'cudnn._Pooling') -function SpatialMaxPooling:__init(kW, kH, dW, dH) - parent.__init(self, kW, kH, dW, dH) +function SpatialMaxPooling:__init(kW, kH, dW, dH, padW, padH) + parent.__init(self, kW, kH, dW, dH, padW, padH) self.mode = 'CUDNN_POOLING_MAX' end |