add _input and _gradOutput to clearState everywhere

1 files changed, 2 insertions, 2 deletions

diff --git a/SpatialSoftMax.lua b/SpatialSoftMax.lua
index f180526..493477f 100644
--- a/SpatialSoftMax.lua
+++ b/SpatialSoftMax.lua
@@ -30,7 +30,7 @@ function SpatialSoftMax:createIODescriptors(input)
       batch = false
    end
    assert(input:dim() == 4 and input:isContiguous());
-   
+
    if not self.iDesc or not self.oDesc or
       input:size(1) ~= self.iSize[1] or input:size(2) ~= self.iSize[2]
    or input:size(3) ~= self.iSize[3] or input:size(4) ~= self.iSize[4] then
@@ -106,6 +106,6 @@ end
 
 function SpatialSoftMax:clearState()
    self:clearDesc()
-   nn.utils.clear(self, '_gradOutput')
+   self._gradOutput = nil
    return parent.clearState(self)
 end