diff options
author | Sergey Zagoruyko <zagoruyko2@gmail.com> | 2016-02-04 17:53:04 +0300 |
---|---|---|
committer | Sergey Zagoruyko <zagoruyko2@gmail.com> | 2016-02-09 21:08:41 +0300 |
commit | b4ebdf2f95ee9f1429825a0d7b0948721e407d82 (patch) | |
tree | 77b6183d0c258d07d5f5e6338c48ccb5fc5a877b /SpatialAdaptiveMaxPooling.lua | |
parent | 4a433463fcd17c4ec3242586da7e9fc859feb3ae (diff) |
nn.clearState
Diffstat (limited to 'SpatialAdaptiveMaxPooling.lua')
-rw-r--r-- | SpatialAdaptiveMaxPooling.lua | 15 |
1 files changed, 9 insertions, 6 deletions
diff --git a/SpatialAdaptiveMaxPooling.lua b/SpatialAdaptiveMaxPooling.lua index 6dfa58b..efe4515 100644 --- a/SpatialAdaptiveMaxPooling.lua +++ b/SpatialAdaptiveMaxPooling.lua @@ -29,11 +29,14 @@ function SpatialAdaptiveMaxPooling:updateGradInput(input, gradOutput) return self.gradInput end +-- for backward compat function SpatialAdaptiveMaxPooling:empty() - self.gradInput:resize() - self.gradInput:storage():resize(0) - self.output:resize() - self.output:storage():resize(0) - self.indices:resize() - self.indices:storage():resize(0) + self:clearState() +end + +function SpatialAdaptiveMaxPooling:clearState() + if self.indices then + self.indices:set() + end + return parent.clearState(self) end |