github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-13 20:39:59 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
b0936e4a44dfb77b0de638d8cd01424dc3ebf614
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