github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/examples/tutorial/new_api/cifar_vit/requirements.txt

6 lines
39 B
Plaintext
Raw Normal View History

[example] add train resnet/vit with booster example (#3694) * [example] add train vit with booster example * [example] update readme * [example] add train resnet with booster example * [example] enable ci * [example] enable ci * [example] add requirements * [hotfix] fix analyzer init * [example] update requirements
2023-05-08 02:42:30 +00:00
colossalai
timm
torch
torchvision
tqdm