ColossalAI/colossalai/zero/gemini
Edenzzzz dcc44aab8d
[misc] Use dist logger in plugins (#6011)
* use dist logger in plugins

* remove trash

* print on rank 0

---------

Co-authored-by: Edenzzzz <wtan45@wisc.edu>
2024-08-20 10:32:41 +08:00
..
chunk [gemini] fixes for benchmarking (#5847) 2024-06-26 15:52:09 +08:00
memory_tracer
__init__.py
gemini_ddp.py [gemini] fixes for benchmarking (#5847) 2024-06-26 15:52:09 +08:00
gemini_hook.py
gemini_mgr.py
gemini_optimizer.py [misc] Use dist logger in plugins (#6011) 2024-08-20 10:32:41 +08:00
placement_policy.py
utils.py