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 /SpatialSubtractiveNormalization.lua | |
| parent | 4a433463fcd17c4ec3242586da7e9fc859feb3ae (diff) | |
nn.clearState
Diffstat (limited to 'SpatialSubtractiveNormalization.lua')
| -rw-r--r-- | SpatialSubtractiveNormalization.lua | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/SpatialSubtractiveNormalization.lua b/SpatialSubtractiveNormalization.lua index 84d943a..e2da2c6 100644 --- a/SpatialSubtractiveNormalization.lua +++ b/SpatialSubtractiveNormalization.lua @@ -106,3 +106,10 @@ function SpatialSubtractiveNormalization:updateGradInput(input, gradOutput) -- done return self.gradInput end + +function SpatialSubtractiveNormalization:clearState() + if self.ones then self.ones:set() end + if self._coef then self._coef:set() end + self.meanestimator:clearState() + return parent.clearState(self) +end |