Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/soumith/cudnn.torch.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorsoumith <soumith@gmail.com>2016-03-17 22:11:17 +0300
committersoumith <soumith@gmail.com>2016-03-17 22:11:17 +0300
commit501dc97f6720f448c8e4fc4fa70829230afe987e (patch)
treeeb07331652c86449204e3ef5ad595acf5ac85c33 /SpatialDivisiveNormalization.lua
parent75b21693b61365cda1499a841f250f165eef6ed1 (diff)
add _input and _gradOutput to clearState everywhere
Diffstat (limited to 'SpatialDivisiveNormalization.lua')
-rw-r--r--SpatialDivisiveNormalization.lua5
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