ColossalAI/colossalai/trainer/hooks
Jiarui Fang 61c20b44bc
[log] local throughput metrics (#811)
* Revert "[zero] add ZeroTensorShardStrategy (#793)"

This reverts commit 88759e289e.

* [gemini] set cpu memory capacity

* [log] local throughput collecting

* polish

* polish

* polish

* polish code

* polish
2022-04-20 10:05:39 +08:00
..
__init__.py
_base_hook.py
_checkpoint_hook.py
_commons_.py [refactor] moving memtracer to gemini (#801) 2022-04-19 10:13:08 +08:00
_log_hook.py [refactor] moving memtracer to gemini (#801) 2022-04-19 10:13:08 +08:00
_lr_scheduler_hook.py
_mem_tracer_hook.py [refactor] moving memtracer to gemini (#801) 2022-04-19 10:13:08 +08:00
_metric_hook.py [log] local throughput metrics (#811) 2022-04-20 10:05:39 +08:00