updating fine-tune classifier documentation (#16063)

This commit is contained in:
David S. Batista 2022-03-10 22:21:56 +01:00 committed by GitHub
parent 6b09328368
commit 96ac7549cb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

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