transformers/tests/deepspeed
Stas Bekman 956a483173
[deepspeed] zero inference (#14253)
* [deepspeed] zero inference

* only z3 makes sense for inference

* fix and style

* docs

* rework

* fix test

* Apply suggestions from code review

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

* responding to suggestions

Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
2021-11-23 14:09:15 -08: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 [deepspeed] zero inference (#14253) 2021-11-23 14:09:15 -08:00
test_model_zoo.py [deepspeed] Enable multiple test runs on single box, defer to DS_TEST_PORT if set (#14331) 2021-11-08 12:40:29 -08:00