transformers/tests/deepspeed
Stas Bekman ce2fef2ad2
[trainer / deepspeed] fix hyperparameter_search (#16740)
* [trainer / deepspeed] fix hyperparameter_search

* require optuna

* style

* oops

* add dep in the right place

* create deepspeed-testing dep group

* Trigger CI
2022-04-14 17:24:38 -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] fix hyperparameter_search (#16740) 2022-04-14 17:24:38 -07:00
test_model_zoo.py [Deepspeed] add support for bf16 mode (#14569) 2022-03-11 17:53:53 -08:00