transformers/tests/deepspeed
Olatunji Ruwase 42f359d015
Use DS callable API to allow hf_scheduler + ds_optimizer (#13216)
* Use DS callable API to allow hf_scheduler + ds_optimizer

* Preserve backward-compatibility

* Restore backward compatibility

* Tweak arg positioning

* Tweak arg positioning

* bump the required version

* Undo indent

* Update src/transformers/trainer.py

* style

Co-authored-by: Stas Bekman <stas@stason.org>
Co-authored-by: Stas Bekman <stas00@users.noreply.github.com>
2021-08-30 10:01:06 -07:00
..
ds_config_zero2.json [Deepspeed] various fixes (#12058) 2021-06-08 08:36:15 -07:00
ds_config_zero3.json [Deepspeed] various fixes (#12058) 2021-06-08 08:36:15 -07:00
test_deepspeed.py Use DS callable API to allow hf_scheduler + ds_optimizer (#13216) 2021-08-30 10:01:06 -07:00
test_model_zoo.py [Deepspeed] adapt multiple models, add zero_to_fp32 tests (#12477) 2021-07-13 12:07:32 -07:00