clearState
This commit is contained in:
parent
2d85f70415
commit
4c8a47e149
|
@ -38,3 +38,9 @@ function AuxiliaryLossTable:updateGradInput(input, gradOutput)
|
||||||
|
|
||||||
return self.gradInput
|
return self.gradInput
|
||||||
end
|
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
|
return self.gradInput
|
||||||
end
|
end
|
||||||
|
function ScaleTable:clearState()
|
||||||
|
self.grad_tmp:set()
|
||||||
|
self.scale:set()
|
||||||
|
return parent:clearstate()
|
||||||
|
end
|
||||||
|
|
Loading…
Reference in a new issue