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 |
ver217
|
8e85d2440a
|
[example] update vit ci script (#2469)
* [example] update vit ci script
* [example] update requirements
* [example] update requirements
|
2023-01-13 13:31:27 +08:00 |