This commit is contained in:
farizrahman4u
2017-03-19 08:43:43 +05:30
parent cb255024f1
commit 71283a211b
+2 -2
View File
@@ -53,8 +53,8 @@ class PELU(Layer):
# Initialised as ones to emulate the default ELU
self.alphas = self.add_weight(param_shape,
name='alpha',
initializer=self.alphas_initializer)
name='alpha',
initializer=self.alphas_initializer)
self.betas = self.add_weight(param_shape, name='betas', initializer=self.betas_initializer)
self.trainable_weights = [self.alphas, self.betas]