wassname/keras-contrib
1
0
Fork 0
mirror of https://github.com/wassname/keras-contrib.git synced 2026-07-29 11:21:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

Bug fix

Browse Source
This commit is contained in:
Fariz Rahman
2017-01-26 14:43:42 +05:30
committed by GitHub
parent 1be552eb6d
commit 40ace94d89
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
keras_contrib/layers/advanced_activations.py
+2 -2
View File
@@ -66,14 +66,14 @@ class PELU(Layer):
def call(self, x, mask=None):
if K.backend() == 'theano':
pos = K.relu(x) * (K.pattern_broadcast(self.alphas, self.param_broadcast) /
pos = x * (K.pattern_broadcast(self.alphas, self.param_broadcast) /
K.pattern_broadcast(self.betas, self.param_broadcast))
neg = (K.pattern_broadcast(self.alphas, self.param_broadcast) *
(K.exp(x / K.pattern_broadcast(self.betas, self.param_broadcast)) - 1))
else:
pos = K.relu(x) * (self.alphas / self.betas)
neg = self.alphas * (K.exp(x / self.betas) - 1)
return pos + neg
return K.switch(x < 0., neg, pos)
def get_config(self):
config = {'alphas_init': self.alphas_init.__name__,