4 Commits (ckpt)

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	1 year ago
Baizhou Zhang	b3ab7fbabf	[example] update ViT example using booster api (#3940)	1 year ago
Jiarui Fang	a4b4bb01d6	[example] update vit readme (#2155)	2 years ago
Jiarui Fang	cf68cc92ac	[example] add vit (#1942) ... * [ColoTensor] ColoInitContext initialize parameters in shard mode. * polish * [example] add vit	2 years ago