From 96ac7549cbb1f1f39f624ad5d52fc07f1f9a2f51 Mon Sep 17 00:00:00 2001 From: "David S. Batista" Date: Thu, 10 Mar 2022 22:21:56 +0100 Subject: [PATCH] updating fine-tune classifier documentation (#16063) --- docs/source/tasks/token_classification.mdx | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/tasks/token_classification.mdx b/docs/source/tasks/token_classification.mdx index 65deb5a31f..fdbf156a3e 100644 --- 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) ```