From 1b11a5d3dda2d10a623ef10e202fba4d41448b69 Mon Sep 17 00:00:00 2001 From: Michael Oliver Date: Fri, 21 Apr 2017 16:47:59 -0700 Subject: [PATCH] make srelu and pelu work with any initializer --- keras_contrib/layers/advanced_activations.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/keras_contrib/layers/advanced_activations.py b/keras_contrib/layers/advanced_activations.py index 22b896a..345fb14 100644 --- a/keras_contrib/layers/advanced_activations.py +++ b/keras_contrib/layers/advanced_activations.py @@ -66,6 +66,7 @@ class PELU(Layer): param_shape[i - 1] = 1 self.param_broadcast[i - 1] = True + param_shape = tuple(param_shape) # Initialised as ones to emulate the default ELU self.alpha = self.add_weight(param_shape, name='alpha', @@ -175,6 +176,8 @@ class SReLU(Layer): param_shape[i - 1] = 1 self.param_broadcast[i - 1] = True + param_shape = tuple(param_shape) + self.t_left = self.add_weight(param_shape, name='t_left', initializer=self.t_left_initializer)