github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-26 17:53:08 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
cea4292ae5ce7563f94bb7deef6f1b9bd155ecc5
ColossalAI/examples/images/vit/configs
History
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
[example] add vit (#1942)
2022-11-14 17:28:03 +08:00