Fix indexing error in dropout rates.

mdenil · Jul 8, 2014 · c99292e · c99292e
1 parent 6f8c362
commit c99292e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mlp.py b/mlp.py
@@ -117,7 +117,7 @@ def __init__(self,
                     input=next_dropout_layer_input,
                     activation=activations[layer_counter],
                     n_in=n_in, n_out=n_out, use_bias=use_bias,
-                    dropout_rate=dropout_rates[layer_counter])
+                    dropout_rate=dropout_rates[layer_counter + 1])
             self.dropout_layers.append(next_dropout_layer)
             next_dropout_layer_input = next_dropout_layer.output