ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI

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