Unverified Commit 96ac7549 authored by David S. Batista's avatar David S. Batista Committed by GitHub
Browse files

updating fine-tune classifier documentation (#16063)

parent 6b093283
......@@ -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>
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment