clearState

lib/AuxiliaryLossTable.lua

@@ -38,3 +38,9 @@ function AuxiliaryLossTable:updateGradInput(input, gradOutput)
 
    return self.gradInput
 end
+function AuxiliaryLossTable:clearState()
+   self.gradInput = {}
+   self.output_table = {}
+   self.output_tensor:set()
+   return parent:clearState()
+end

lib/ScaleTable.lua

@@ -32,3 +32,8 @@ function ScaleTable:updateGradInput(input, gradOutput)
 
    return self.gradInput
 end
+function ScaleTable:clearState()
+   self.grad_tmp:set()
+   self.scale:set()
+   return parent:clearstate()
+end