1
0
Fork 0
mirror of synced 2024-07-01 20:50:29 +12:00

Add support for clearState() in w2nn.LeakyReLU

This commit is contained in:
nagadomi 2016-03-12 06:54:09 +09:00
parent 1900ac7500
commit 223dcead67

View file

@ -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