mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-08-14 22:23:23 +00:00
[example] update vit readme (#2155)
This commit is contained in:
parent
2cfe685b9f
commit
a4b4bb01d6
@ -13,7 +13,7 @@ We use model **vision_transformer** from timm [Link](https://github.com/rwightma
|
|||||||
|
|
||||||
# Requirement
|
# Requirement
|
||||||
|
|
||||||
You should install colossalai from main branch with commit 561e904.
|
Install colossalai version >= 0.1.11
|
||||||
|
|
||||||
## Unit test
|
## Unit test
|
||||||
To run unit test, you should install pytest, transformers with:
|
To run unit test, you should install pytest, transformers with:
|
||||||
|
Loading…
Reference in New Issue
Block a user