Fix dropout bug

Signed-off-by: begeekmyfriend <[email protected]>
begeekmyfriend · Sep 2, 2018 · e7e1ee5 · e7e1ee5
1 parent 1f8d32f
commit e7e1ee5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/modules.py b/models/modules.py
@@ -8,7 +8,7 @@ def prenet(inputs, is_training, layer_sizes, scope=None):
   with tf.variable_scope(scope or 'prenet'):
     for i, size in enumerate(layer_sizes):
       dense = tf.layers.dense(x, units=size, activation=tf.nn.relu, name='dense_%d' % (i+1))
-      x = tf.layers.dropout(dense, rate=drop_rate, name='dropout_%d' % (i+1))
+      x = tf.layers.dropout(dense, rate=drop_rate, training=is_training, name='dropout_%d' % (i+1))
   return x