diff options
author | soumith <soumith@gmail.com> | 2016-03-17 22:11:17 +0300 |
---|---|---|
committer | soumith <soumith@gmail.com> | 2016-03-17 22:11:17 +0300 |
commit | 501dc97f6720f448c8e4fc4fa70829230afe987e (patch) | |
tree | eb07331652c86449204e3ef5ad595acf5ac85c33 /SpatialDivisiveNormalization.lua | |
parent | 75b21693b61365cda1499a841f250f165eef6ed1 (diff) |
add _input and _gradOutput to clearState everywhere
Diffstat (limited to 'SpatialDivisiveNormalization.lua')
-rw-r--r-- | SpatialDivisiveNormalization.lua | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/SpatialDivisiveNormalization.lua b/SpatialDivisiveNormalization.lua index 50d7923..3462b61 100644 --- a/SpatialDivisiveNormalization.lua +++ b/SpatialDivisiveNormalization.lua @@ -98,3 +98,8 @@ function DivisiveNorm:write(f) end f:writeObject(var) end + +function DivisiveNorm:clearState() + self._gradOutput = nil + return parent.clearState(self) +end |