Fixed Style Inconsistency (#3976)

This commit is contained in:
Jordan 2020-04-30 07:33:09 -05:00 committed by GitHub
parent 64070cbb88
commit 7f9193ef09
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

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