CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-06-29 11:41:29 +12:00
Code Issues Releases Wiki Activity

Add support for clearState() in w2nn.LeakyReLU

Browse source
This commit is contained in:
nagadomi 2016-03-12 06:54:09 +09:00
parent 1900ac7500
commit 223dcead67
1 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
lib/LeakyReLU.lua
View file

@ -17,7 +17,7 @@ function LeakyReLU:updateOutput(input)
return self.output return self.output
end end
function LeakyReLU:updateGradInput(input, gradOutput) function LeakyReLU:updateGradInput(input, gradOutput)
self.gradInput:resizeAs(gradOutput) self.gradInput:resizeAs(gradOutput)
-- filter positive -- filter positive
@ -29,3 +29,8 @@ function LeakyReLU:updateGradInput(input, gradOutput)
return self.gradInput return self.gradInput
end end
function LeakyReLU:clearState()
nn.utils.clear(self, 'negative')
return parent.clearState(self)
end