transformers/tests/deepspeed
Stas Bekman 7ec596ecda
[DeepSpeed] decouple DeepSpeedConfigHF from Trainer (#11966)
* decouple DeepSpeedConfigHF from Trainer

* add LoggingLevel ctx manager; add new test

* cleanup

* add docs

* Apply suggestions from code review

Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>

* implemented suggested renames

* formatter workaround

Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
2021-06-01 13:24:52 -07:00
..
ds_config_zero2.json [Deepspeed] ZeRO-Infinity integration plus config revamp (#11418) 2021-04-26 10:40:32 -07:00
ds_config_zero3.json [Deepspeed] ZeRO-Infinity integration plus config revamp (#11418) 2021-04-26 10:40:32 -07:00
test_deepspeed.py [DeepSpeed] decouple DeepSpeedConfigHF from Trainer (#11966) 2021-06-01 13:24:52 -07:00