transformers/tests/deepspeed
Stas Bekman d28b7aa8cb
[deepspeed / testing] reset global state (#17553)
* [deepspeed] fix load_best_model test

* [deepspeed] add state reset on unittest tearDown
2022-06-06 07:49:25 -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 [deepspeed / testing] reset global state (#17553) 2022-06-06 07:49:25 -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