File size: 181 Bytes
8bdcd41 |
1 2 3 4 5 |
from transformers import BertForSequenceClassification
model = BertForSequenceClassification.from_pretrained('bert-base-uncased', num_labels=2)
model.save_pretrained('./hf_model')
|
8bdcd41 |
1 2 3 4 5 |
from transformers import BertForSequenceClassification
model = BertForSequenceClassification.from_pretrained('bert-base-uncased', num_labels=2)
model.save_pretrained('./hf_model')
|