diff options
author | Alfredo Canziani <alfredo.canziani@gmail.com> | 2016-03-02 06:12:12 +0300 |
---|---|---|
committer | Alfredo Canziani <alfredo.canziani@gmail.com> | 2016-03-02 06:12:12 +0300 |
commit | fadfadb961e492f0da22e95386e96c1bffff019c (patch) | |
tree | abb3921d29fc15b91b7a9f9d8a3192db5ba761cb /SpatialAveragePooling.lua | |
parent | 260cc314fe5336b1d8bff56294a135a052514d7c (diff) |
Better __tostring__ and cleans formatting
Diffstat (limited to 'SpatialAveragePooling.lua')
-rw-r--r-- | SpatialAveragePooling.lua | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/SpatialAveragePooling.lua b/SpatialAveragePooling.lua index 231177d..1e76668 100644 --- a/SpatialAveragePooling.lua +++ b/SpatialAveragePooling.lua @@ -83,11 +83,11 @@ function SpatialAveragePooling:updateGradInput(input, gradOutput) end function SpatialAveragePooling:__tostring__() - local s = string.format('%s(%d,%d,%d,%d', torch.type(self), + local s = string.format('%s(%dx%d, %d,%d', torch.type(self), self.kW, self.kH, self.dW, self.dH) if (self.padW or self.padH) and (self.padW ~= 0 or self.padH ~= 0) then - s = s .. ',' .. self.padW .. ','.. self.padH + s = s .. ', ' .. self.padW .. ','.. self.padH end s = s .. ')' - return s + return s end |