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

Adopt framework-specific blocks for content (#16342)

Browse Source 
* ✨ refactor code samples with framework-specific blocks

* ✨ update training.mdx

* 🖍 apply feedback
This commit is contained in:
Steven Liu
2022-03-22 14:14:58 -07:00
committed by GitHub
parent 62cbd8423b
commit 7732148124
13 changed files with 169 additions and 133 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
docs/source/tasks/image_classification.mdx
View File

@@ -109,8 +109,10 @@ Use [`DefaultDataCollator`] to create a batch of examples. Unlike other data col
>>> data_collator = DefaultDataCollator()
```
## Fine-tune with Trainer
## Train
<frameworkcontent>
<pt>
Load ViT with [`AutoModelForImageClassification`]. Specify the number of labels, and pass the model the mapping between label number and label class:
```py
@@ -162,6 +164,8 @@ At this point, only three steps remain:
>>> trainer.train()
```
</pt>
</frameworkcontent>
<Tip>