SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

updating fine-tune classifier documentation (#16063)

Browse Source
This commit is contained in:
David S. Batista
2022-03-10 22:21:56 +01:00
committed by GitHub
parent 6b09328368
commit 96ac7549cb
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docs/source/tasks/token_classification.mdx
View File

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