mirror of
https://github.com/huggingface/transformers.git
synced 2025-08-02 11:11:05 +06:00
updating fine-tune classifier documentation (#16063)
This commit is contained in:
parent
6b09328368
commit
96ac7549cb
@ -151,7 +151,7 @@ Load DistilBERT with [`AutoModelForTokenClassification`] along with the number o
|
||||
```py
|
||||
>>> from transformers import AutoModelForTokenClassification, TrainingArguments, Trainer
|
||||
|
||||
>>> model = AutoModelForTokenClassification.from_pretrained("distilbert-base-uncased", num_labels=2)
|
||||
>>> model = AutoModelForTokenClassification.from_pretrained("distilbert-base-uncased", num_labels=14)
|
||||
```
|
||||
|
||||
<Tip>
|
||||
|
Loading…
Reference in New Issue
Block a user