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