Fix for invalid dropout values

chardmeier · Apr 12, 2016 · d50f469 · d50f469
1 parent 26714bc
commit d50f469
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/keras/layers/core.py b/keras/layers/core.py
@@ -76,9 +76,10 @@ class Dropout(Layer):
         - [Dropout: A Simple Way to Prevent Neural Networks from Overfitting](http://www.cs.toronto.edu/~rsalakhu/papers/srivastava14a.pdf)
     '''
     def __init__(self, p, **kwargs):
-        self.supports_masking = True
-        self.uses_learning_phase = True
         self.p = p
+        if 0. < self.p < 1.:
+            self.uses_learning_phase = True
+        self.supports_masking = True
         super(Dropout, self).__init__(**kwargs)
 
     def call(self, x, mask=None):