2 Commits (608cffaed3821bacdfce7c44cdf09e6cd38d32c2)

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