fix mobilebert

zheyuye · Jun 29, 2020 · aea936f · aea936f
1 parent eead164
commit aea936f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/gluonnlp/models/mobilebert.py b/src/gluonnlp/models/mobilebert.py
@@ -343,6 +343,7 @@ def __init__(self,
                                                num_heads=num_heads,
                                                attention_dropout_prob=attention_dropout_prob,
                                                hidden_dropout_prob=hidden_dropout_prob,
+                                               activation_dropout_prob=hidden_dropout_prob,
                                                num_stacked_ffn=num_stacked_ffn,
                                                bottleneck_strategy=bottleneck_strategy,
                                                layer_norm_eps=layer_norm_eps,