ColossalAI/examples/images/vit/configs
Jiarui Fang cf68cc92ac
[example] add vit (#1942)
* [ColoTensor] ColoInitContext initialize parameters in shard mode.

* polish

* [example] add vit
2022-11-14 17:28:03 +08:00
..
vit_1d_tp2.py