Fixed Style Inconsistency (#3976)

2020-04-30 07:33:09 -05:00
parent 64070cbb88
commit 7f9193ef09
1 changed files with 1 additions and 1 deletions
--- a/src/transformers/modeling_bert.py
+++ b/src/transformers/modeling_bert.py
@@ -1079,7 +1079,7 @@ class BertForSequenceClassification(BertPreTrainedModel):

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

        self.init_weights()