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