ColossalAI

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
ver217	8e85d2440a	[example] update vit ci script (#2469 ) * [example] update vit ci script * [example] update requirements * [example] update requirements	2023-01-13 13:31:27 +08:00