ColossalAI/examples/tutorial/new_api/cifar_resnet/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

5 lines
34 B
Plaintext

colossalai
torch
torchvision
tqdm