clearState

1 files changed, 10 insertions, 1 deletions

diff --git a/TemporalConvolution.lua b/TemporalConvolution.lua
index 3e646e5..72a87c3 100644
--- a/TemporalConvolution.lua
+++ b/TemporalConvolution.lua
@@ -104,10 +104,14 @@ function TemporalConvolution:accGradParameters(input,gradOutput,scale)
     cudnn.SpatialConvolution.accGradParameters(self,_input,_gradOutput,scale)
 end
 
-function TemporalConvolution:write(f)
+function TemporalConvolution:clearDesc()
   self.buffer = nil
   self._ouptut = nil
   self.oSize = nil
+end
+
+function TemporalConvolution:write(f)
+  self:clearDesc()
   cudnn.SpatialConvolution.clearDesc(self)
     local var = {}
     for k,v in pairs(self) do
@@ -115,3 +119,8 @@ function TemporalConvolution:write(f)
     end
     f:writeObject(var)
 end
+
+function TemporalConvolution:clearState()
+   self:clearDesc()
+   return parent.clearState(self)
+end