diff --git a/src/transformers/models/bert/modeling_bert.py b/src/transformers/models/bert/modeling_bert.py index 0e32d04370b..60cf7ae26ff 100755 --- 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.classifier_dropout_prob) + self.dropout = nn.Dropout(config.classifier_dropout) self.classifier = nn.Linear(config.hidden_size, 1) self.init_weights() diff --git a/src/transformers/models/mobilebert/modeling_mobilebert.py b/src/transformers/models/mobilebert/modeling_mobilebert.py index 9b2a8661e5f..c9c59277963 100644 --- a/src/transformers/models/mobilebert/modeling_mobilebert.py +++ b/src/transformers/models/mobilebert/modeling_mobilebert.py @@ -1415,7 +1415,7 @@ class MobileBertForMultipleChoice(MobileBertPreTrainedModel): super().__init__(config) self.mobilebert = MobileBertModel(config) - self.dropout = nn.Dropout(config.hidden_dropout_prob) + self.dropout = nn.Dropout(config.classifier_dropout) self.classifier = nn.Linear(config.hidden_size, 1) self.init_weights()