fix to load model saved from old keras version

akeshavan · Feb 3, 2018 · 10d6a2b · 10d6a2b
1 parent ad00676
commit 10d6a2b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/keras/layers/advanced_activations.py b/keras/layers/advanced_activations.py
@@ -40,7 +40,11 @@ class LeakyReLU(Layer):
     def __init__(self, alpha=0.3, **kwargs):
         super(LeakyReLU, self).__init__(**kwargs)
         self.supports_masking = True
-        self.alpha = K.cast_to_floatx(alpha)
+        # fix from https://github.com/keras-team/keras/pull/7784
+        if isinstance(alpha, dict):
+            self.alpha = K.cast_to_floatx(alpha['value'])
+        else:
+            self.alpha = K.cast_to_floatx(alpha)
 
     def call(self, inputs):
         return K.relu(inputs, alpha=self.alpha)