3 Commits (main)

Author	SHA1	Message	Date
Baizhou Zhang	df66741f77	[bug] fix get_default_parser in examples (#4764)	2023-09-21 10:42:25 +08:00
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