mirror of https://github.com/hpcaitech/ColossalAI
1f894e033f
* 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 |
||
---|---|---|
.. | ||
__init__.py | ||
memory_monitor.py | ||
memstats_collector.py | ||
model_data_memtracer.py |