transformers/examples/tests/deepspeed
Stas Bekman 77b862847b
[DeepSpeed] restore memory for evaluation (#10114)
* free up memory at the end of train

* rework tests

* consistent formatting

* correction
2021-02-10 09:09:48 -08:00
..
ds_config.json [DeepSpeed] restore memory for evaluation (#10114) 2021-02-10 09:09:48 -08:00
test_deepspeed.py [DeepSpeed] restore memory for evaluation (#10114) 2021-02-10 09:09:48 -08:00