CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-06-26 18:20:26 +12:00
Code Issues Releases Wiki Activity

Fix clearState

Browse source
This commit is contained in:
nagadomi 2016-03-28 18:38:01 +09:00
parent 5ed13d6c1b
commit fd06a47d56
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
lib/srcnn.lua
View file

@ -19,10 +19,10 @@ end
function nn.SpatialConvolutionMM:clearState()
if self.gradWeight then
self.gradWeight = torch.Tensor(self.nOutputPlane, self.nInputPlane * self.kH * self.kW):typeAs(self.gradWeight):zero()
self.gradWeight:resize(self.nOutputPlane, self.nInputPlane * self.kH * self.kW):zero()
end
if self.gradBias then
self.gradBias = torch.Tensor(self.nOutputPlane):typeAs(self.gradBias):zero()
self.gradBias:resize(self.nOutputPlane):zero()
end
return nn.utils.clear(self, 'finput', 'fgradInput', '_input', '_gradOutput', 'output', 'gradInput')
end