diff options
Diffstat (limited to 'SpatialSoftMax.lua')
-rw-r--r-- | SpatialSoftMax.lua | 4 |
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 |