ColossalAI/colossalai/gemini/memory_tracer
Jiarui Fang c5d39215f6
Revert "[feature] new zero implementation (#1623)" (#1643)
This reverts commit 5be118f405.
2022-09-26 10:06:03 +08:00
..
__init__.py [refactor] moving memtracer to gemini (#801) 2022-04-19 10:13:08 +08:00
memory_monitor.py [refactor] moving memtracer to gemini (#801) 2022-04-19 10:13:08 +08:00
memstats_collector.py Revert "[feature] new zero implementation (#1623)" (#1643) 2022-09-26 10:06:03 +08:00
model_data_memtracer.py [refactor] moving memtracer to gemini (#801) 2022-04-19 10:13:08 +08:00