From ed6cc315ef6672c1f2824a0ffe1c23bff7febe8a Mon Sep 17 00:00:00 2001 From: 1FengL Date: Tue, 21 May 2019 14:21:31 +0100 Subject: [PATCH] fix dynamic model cannot track PRelu weights gradients problem --- tensorlayer/layers/activation.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tensorlayer/layers/activation.py b/tensorlayer/layers/activation.py index 44fcc47a9..0d50b4a4a 100644 --- a/tensorlayer/layers/activation.py +++ b/tensorlayer/layers/activation.py @@ -95,6 +95,7 @@ def build(self, inputs_shape): def forward(self, inputs): pos = tf.nn.relu(inputs) + self.alpha_var_constrained = tf.nn.sigmoid(self.alpha_var, name="constraining_alpha_var_in_0_1") neg = -self.alpha_var_constrained * tf.nn.relu(-inputs) return pos + neg