various fixes

This commit is contained in:
thomwolf committed 2018-11-02 01:52:54 +01:00
1 parent 2c731fd129
commit dee09a40b2
2 files changed
+11 -8

No files matched your search

+2 -1
View File
@@ -412,7 +412,8 @@ class BertForSequenceClassification(nn.Module):
model = modeling.BertModel(config, num_labels)
logits = model(input_ids, token_type_ids, input_mask)
```
""" def __init__(self, config, num_labels):
"""
def __init__(self, config, num_labels):
super(BertForSequenceClassification, self).__init__()
self.bert = BertModel(config)
self.dropout = nn.Dropout(config.hidden_dropout_prob)