github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-01 17:17:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

[example] update vit readme (#2155)

Browse Source
This commit is contained in:
Jiarui Fang
2022-12-20 15:56:54 +08:00
committed by GitHub
parent 2cfe685b9f
commit a4b4bb01d6
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/images/vit/README.md
View File

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