From 2fd8a357d912f89e628f3db93336a34376570b1d Mon Sep 17 00:00:00 2001 From: soumith Date: Thu, 17 Mar 2016 14:33:20 -0400 Subject: adding _input and _gradOutput to clearState --- SpatialConvolution.lua | 2 ++ 1 file changed, 2 insertions(+) (limited to 'SpatialConvolution.lua') diff --git a/SpatialConvolution.lua b/SpatialConvolution.lua index 09e421d..3be7715 100644 --- a/SpatialConvolution.lua +++ b/SpatialConvolution.lua @@ -469,5 +469,7 @@ end function SpatialConvolution:clearState() self:clearDesc() + self._input = nil + self._gradOutput = nil return nn.Module.clearState(self) end -- cgit v1.2.3