search

hpcaitech / ColossalAI-Examples

Examples of training models with hybrid parallelism using ColossalAI
Apache License 2.0
334 stars 102 forks source link

[example]support ViT with ColoTensor #141

Closed ZhaoYi1222 closed 2 years ago