ColossalAI/examples/tutorial/new_api/cifar_vit/requirements.txt
Hongxin Liu f83ea813f5
[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 10:42:30 +08:00

6 lines
39 B
Plaintext

colossalai
timm
torch
torchvision
tqdm