Stas Bekman 2f59ad1609 [trainer/deepspeed] load_best_model (reimplement re-init) (#17151) ...

* [trainer/deepspeed] load_best_model

* to sync with DS PR #1947

* simplify

* rework load_best_model test

* cleanup

* bump deepspeed>=0.6.5

Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>

2022-06-02 09:14:21 -07:00

..

ds_config_zero2.json

[Deepspeed] add support for bf16 mode (#14569)

2022-03-11 17:53:53 -08:00

ds_config_zero3.json

[Deepspeed] add support for bf16 mode (#14569)

2022-03-11 17:53:53 -08:00

test_deepspeed.py

[trainer/deepspeed] load_best_model (reimplement re-init) (#17151)

2022-06-02 09:14:21 -07:00

test_model_zoo.py

[Deepspeed] add many more models to the model zoo test (#12695)

2022-05-10 08:22:42 -07:00

vit_feature_extractor.json

missing file (#17164)

2022-05-10 10:19:50 -07:00