use nn.utils.clear

8 files changed, 8 insertions, 12 deletions

diff --git a/Pointwise.lua b/Pointwise.lua
index 0a79545..8d0a06e 100644
--- a/Pointwise.lua
+++ b/Pointwise.lua
@@ -69,6 +69,6 @@ end
 
 function Pointwise:clearState()
    self:clearDesc()
-   self._gradOutput = nil
+   nn.utils.clear(self, '_gradOutput')
    return parent.clearState(self)
 end
diff --git a/Pooling.lua b/Pooling.lua
index c2061e3..00bd0a8 100644
--- a/Pooling.lua
+++ b/Pooling.lua
@@ -132,6 +132,6 @@ end
 
 function Pooling:clearState()
    self:clearDesc()
-   self._gradOutput = nil
+   nn.utils.clear(self, '_gradOutput')
    return parent.clearState(self)
 end
diff --git a/Pooling3D.lua b/Pooling3D.lua
index 072f2c3..489865c 100644
--- a/Pooling3D.lua
+++ b/Pooling3D.lua
@@ -141,6 +141,6 @@ end
 
 function Pooling:clearState()
    self:clearDesc()
-   self._gradOutput = nil
+   nn.utils.clear(self, '_gradOutput')
    return parent.clearState(self)
 end
diff --git a/SpatialConvolution.lua b/SpatialConvolution.lua
index 71eaa69..20f31ef 100644
--- a/SpatialConvolution.lua
+++ b/SpatialConvolution.lua
@@ -466,7 +466,6 @@ end
 
 function SpatialConvolution:clearState()
    self:clearDesc()
-   self._input = nil
-   self._gradOutput = nil
+   nn.utils.clear(self, '_input', '_gradOutput')
    return nn.Module.clearState(self)
 end
diff --git a/SpatialCrossMapLRN.lua b/SpatialCrossMapLRN.lua
index aefe618..f6e7cd9 100644
--- a/SpatialCrossMapLRN.lua
+++ b/SpatialCrossMapLRN.lua
@@ -106,6 +106,6 @@ end
 
 function LRN:clearState()
    self:clearDesc()
-   self._gradOutput = nil
+   nn.utils.clear(self, '_gradOutput')
    return nn.Module.clearState(self)
 end
diff --git a/SpatialSoftMax.lua b/SpatialSoftMax.lua
index 493477f..4b3a488 100644
--- a/SpatialSoftMax.lua
+++ b/SpatialSoftMax.lua
@@ -106,6 +106,6 @@ end
 
 function SpatialSoftMax:clearState()
    self:clearDesc()
-   self._gradOutput = nil
+   nn.utils.clear(self, '_gradOutput')
    return parent.clearState(self)
 end
diff --git a/TemporalConvolution.lua b/TemporalConvolution.lua
index a9e6470..36a55b7 100644
--- a/TemporalConvolution.lua
+++ b/TemporalConvolution.lua
@@ -122,7 +122,6 @@ end
 
 function TemporalConvolution:clearState()
    self:clearDesc()
-   self._gradOutput = nil
-   self._input = nil
+   nn.utils.clear(self, '_input', '_gradOutput')
    return parent.clearState(self)
 end
diff --git a/VolumetricConvolution.lua b/VolumetricConvolution.lua
index 1f8025c..6ec3302 100644
--- a/VolumetricConvolution.lua
+++ b/VolumetricConvolution.lua
@@ -307,8 +307,6 @@ end
 
 function VolumetricConvolution:clearState()
    self:clearDesc()
-   nn.utils.clear(self, 'extraBuffer')
-   self._gradOutput = nil
-   self._input = nil
+   nn.utils.clear(self, 'extraBuffer', '_input', '_gradOutput')
    return nn.Module.clearState(self)
 end