diff options
| -rw-r--r-- | SpatialSubtractiveNormalization.lua | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/SpatialSubtractiveNormalization.lua b/SpatialSubtractiveNormalization.lua index 7fa4402..d92614a 100644 --- a/SpatialSubtractiveNormalization.lua +++ b/SpatialSubtractiveNormalization.lua @@ -109,6 +109,7 @@ function SpatialSubtractiveNormalization:updateGradInput(input, gradOutput) end function SpatialSubtractiveNormalization:clearState() + self._inpsz = nil if self.ones then self.ones:set() end if self._coef then self._coef:set() end self.meanestimator:clearState() |