diff options
author | Sergey Zagoruyko <zagoruyko2@gmail.com> | 2015-03-04 17:39:05 +0300 |
---|---|---|
committer | Sergey Zagoruyko <zagoruyko2@gmail.com> | 2015-03-04 17:39:05 +0300 |
commit | cb303329bd82f1bb80d3c7eebc14d41195acec83 (patch) | |
tree | cf3fc5fcafe6392e5c11e657c3809e9e070de855 /SpatialLPPooling.lua | |
parent | 7470cc81d458066d751156df52d05d3482411e6c (diff) |
SpatialAveragePooling divides by kW*kH
Diffstat (limited to 'SpatialLPPooling.lua')
-rw-r--r-- | SpatialLPPooling.lua | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/SpatialLPPooling.lua b/SpatialLPPooling.lua index 4e6fe74..fc56296 100644 --- a/SpatialLPPooling.lua +++ b/SpatialLPPooling.lua @@ -17,6 +17,7 @@ function SpatialLPPooling:__init(nInputPlane, pnorm, kW, kH, dW, dH) self:add(nn.Power(pnorm)) end self:add(nn.SpatialAveragePooling(kW, kH, dW, dH)) + self:add(nn.MulConstant(kW*kH)) if pnorm == 2 then self:add(nn.Sqrt()) else |