updating fine-tune classifier documentation (#16063)

2022-03-10 22:21:56 +01:00
parent 6b09328368
commit 96ac7549cb
1 changed files with 1 additions and 1 deletions
--- a/docs/source/tasks/token_classification.mdx
+++ b/docs/source/tasks/token_classification.mdx
@@ -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>