ColossalAI/colossalai/gemini/memory_tracer
Jiarui Fang 223332ff7e
[Gemini] rename ParamTracerWrapper -> RuntimeMemTracer (#2073)
2022-12-05 12:45:11 +08:00
..
__init__.py [Gemini] remove not used MemtracerWrapper (#2072) 2022-12-05 11:57:59 +08:00
chunk_memstats_collector.py [Gemini] polish memstats collector (#1962) 2022-11-16 15:45:57 +08:00
memory_monitor.py [Gemini] independent runtime tracer (#1974) 2022-11-18 10:53:42 +08:00
memstats_collector.py [Gemini] polish memstats collector (#1962) 2022-11-16 15:45:57 +08:00
model_data_memtracer.py [Gemini] fix grad unreleased issue and param recovery issue (#2052) 2022-12-02 16:04:19 +08:00
runtime_mem_tracer.py [Gemini] rename ParamTracerWrapper -> RuntimeMemTracer (#2073) 2022-12-05 12:45:11 +08:00
static_memstats_collector.py [Gemini] polish memstats collector (#1962) 2022-11-16 15:45:57 +08:00