github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-05 12:24:38 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
ea0f6b8df9d8c07480fb146cc2a9dbfd8a7dc406
ColossalAI/examples/images/vit/test_vit.py
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

5.5 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink