diff --git a/src/transformers/modeling_bert.py b/src/transformers/modeling_bert.py index dafcaf3689b..3e409cfb742 100644 --- 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()