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