From 5eaa736fee9af57eaf2e1ba05ecf5ec0d30b05fc Mon Sep 17 00:00:00 2001 From: Michael Oliver Date: Sat, 29 Jul 2017 12:57:45 -0700 Subject: [PATCH] change np.exp to K.exp To address https://github.com/farizrahman4u/keras-contrib/issues/127 --- keras_contrib/layers/normalization.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/keras_contrib/layers/normalization.py b/keras_contrib/layers/normalization.py index 6f8cbbd..a9aba89 100644 --- a/keras_contrib/layers/normalization.py +++ b/keras_contrib/layers/normalization.py @@ -319,8 +319,8 @@ class BatchRenormalization(Layer): # update r_max and d_max t_val = K.get_value(self.t) - r_val = self.r_max_value / (1 + (self.r_max_value - 1) * np.exp(-t_val)) - d_val = self.d_max_value / (1 + ((self.d_max_value / 1e-3) - 1) * np.exp(-(2 * t_val))) + r_val = self.r_max_value / (1 + (self.r_max_value - 1) * K.exp(-t_val)) + d_val = self.d_max_value / (1 + ((self.d_max_value / 1e-3) - 1) * K.exp(-(2 * t_val))) t_val += float(self.t_delta) self.add_update([K.update(self.r_max, r_val),