Skip to content

GitLab

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
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
docs/source/tasks/token_classification.mdx
View file @ 96ac7549
...@@ -151,7 +151,7 @@ Load DistilBERT with [`AutoModelForTokenClassification`] along with the number o ...@@ -151,7 +151,7 @@ Load DistilBERT with [`AutoModelForTokenClassification`] along with the number o
```py ```py
>>> from transformers import AutoModelForTokenClassification, TrainingArguments, Trainer >>> 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> <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