ColossalAI/colossalai/gemini/memory_tracer
ver217 1f894e033f
[gemini] zero supports gemini (#1093)
* add placement policy

* add gemini mgr

* update mem stats collector

* update zero

* update zero optim

* fix bugs

* zero optim monitor os

* polish unit test

* polish unit test

* add assert
2022-06-10 14:48:28 +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 [gemini] zero supports gemini (#1093) 2022-06-10 14:48:28 +08:00
model_data_memtracer.py [refactor] moving memtracer to gemini (#801) 2022-04-19 10:13:08 +08:00