clearState
This commit is contained in:
parent
2d85f70415
commit
4c8a47e149
|
@ -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
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Reference in a new issue