ColossalAI/colossalai/trainer
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
..
hooks [log] local throughput metrics (#811) 2022-04-20 10:05:39 +08:00
__init__.py Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
_trainer.py [refactor] pipeline, put runtime schedule into engine. (#627) 2022-04-03 20:46:45 +08:00