Explorar o código

Add support for clearState() in w2nn.LeakyReLU

nagadomi %!s(int64=9) %!d(string=hai) anos
pai
achega
223dcead67
Modificáronse 1 ficheiros con 6 adicións e 1 borrados
  1. 6 1
      lib/LeakyReLU.lua

+ 6 - 1
lib/LeakyReLU.lua

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