transformers/examples/tests/deepspeed
Stas Bekman 97e688bc22
[Trainer] memory tracker metrics (#10225)
* memory tracker metrics

* go back to eval for somewhat consistency

* handle no-gpu case

* deal with stackable eval calls

* restore callback order

* style

* simplify the API

* add test

* docs

* consistently use eval_ prefix

* improve docs

* Update src/transformers/trainer_utils.py

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

* rename method

* style

Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
2021-02-18 09:27:32 -08:00
..
ds_config.json [DeepSpeed] restore memory for evaluation (#10114) 2021-02-10 09:09:48 -08:00
test_deepspeed.py [Trainer] memory tracker metrics (#10225) 2021-02-18 09:27:32 -08:00