diff options
| author | soumith <soumith@fb.com> | 2014-11-26 06:11:14 +0300 |
|---|---|---|
| committer | soumith <soumith@fb.com> | 2014-11-26 06:12:14 +0300 |
| commit | 875067f4eb6f5c8eab77ee1acd030fd5e2225fc5 (patch) | |
| tree | ebfe4a97600a40947dfb0a875ea67840f8c0d81c /Pooling.lua | |
| parent | b0e6e3f0ad10e931a1f83197b127a442e179e67e (diff) | |
lint fixes
Diffstat (limited to 'Pooling.lua')
| -rw-r--r-- | Pooling.lua | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/Pooling.lua b/Pooling.lua index 94c616a..97eb617 100644 --- a/Pooling.lua +++ b/Pooling.lua @@ -46,8 +46,12 @@ function Pooling:createIODescriptors(input) self.iDesc = cudnn.toDescriptor(input) self.oDesc = cudnn.toDescriptor(self.output) if not batch then - self.gradInput = self.gradInput:view(self.gradInput:size(2), self.gradInput:size(3), self.gradInput:size(4)) - self.output = self.output:view(self.output:size(2), self.output:size(3), self.output:size(4)) + self.gradInput = self.gradInput:view(self.gradInput:size(2), + self.gradInput:size(3), + self.gradInput:size(4)) + self.output = self.output:view(self.output:size(2), + self.output:size(3), + self.output:size(4)) end end end |