From d50f469c09ba48cfb850556699390e6c50e4e651 Mon Sep 17 00:00:00 2001 From: fchollet Date: Tue, 12 Apr 2016 09:41:29 -0700 Subject: [PATCH] Fix for invalid dropout values --- keras/layers/core.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/keras/layers/core.py b/keras/layers/core.py index eae9e7eeb88..b7c20c4f7bc 100644 --- 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):