Update modeling_bert.py (#13129)

7d2feb3a · weierstrass_walker · GitHub · a13c8145 · 7d2feb3a
Unverified Commit 7d2feb3a authored Aug 16, 2021 by weierstrass_walker Committed by GitHub Aug 16, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/models/bert/modeling_bert.py src/transformers/models/bert/modeling_bert.py +1 -1

No files found.
--- a/src/transformers/models/bert/modeling_bert.py
+++ b/src/transformers/models/bert/modeling_bert.py
@@ -1585,7 +1585,7 @@ class BertForMultipleChoice(BertPreTrainedModel):
        super().__init__(config)

        self.bert = BertModel(config)
-        self.dropout = nn.Dropout(config.hidden_dropout_prob)
+        self.dropout = nn.Dropout(config.classifier_dropout_prob)
        self.classifier = nn.Linear(config.hidden_size, 1)

        self.init_weights()