don't require the user to override every layer
@@ -570,6 +570,7 @@ class Dropout(Layer):
include_prob = self.include_prob
if dropout_dict is not None:
assert dropout
+ if self.name in dropout_dict:
include_prob = dropout_dict[self.name]
if dropout:
return tf.nn.dropout(x, include_prob)
