riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

Update modeling_bert.py (#13129)

Browse Source
This commit is contained in:
weierstrass_walker 2021-08-16 14:17:37 +06:00 committed by GitHub
parent a13c8145bc
commit 7d2feb3a3b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/models/bert/modeling_bert.py
View File

@ -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()