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
..
diffusion	[example] update streamlit 0.73.1 to 1.11.1 (#4386)	2023-08-30 22:54:45 +08:00
dreambooth	[gemini] improve compatibility and add static placement policy (#4479)	2023-08-24 09:29:25 +08:00
resnet	[gemini] improve compatibility and add static placement policy (#4479)	2023-08-24 09:29:25 +08:00
vit	[example] update vit example for hybrid parallel plugin (#4641)	2023-09-07 17:38:45 +08:00