github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-04 10:34:41 +00:00
Code Issues Packages Projects Releases Wiki Activity
2,767 Commits 59 Branches 51 Tags
cloud/coati
Commit Graph

2 Commits

Author SHA1 Message Date
Baizhou Zhang
295b38fecf [example] update vit example for hybrid parallel plugin (#4641) 
* update vit example for hybrid plugin

* reset tp/pp size

* fix dataloader iteration bug

* update optimizer passing in evaluation/add grad_accum

* change criterion

* wrap tqdm

* change grad_accum to grad_checkpoint

* fix pbar
 2023-09-07 17:38:45 +08:00
Baizhou Zhang
b3ab7fbabf [example] update ViT example using booster api (#3940) 2023-06-12 15:02:27 +08:00