ColossalAI/docs/colossalai/colossalai.utils.memory_tracer.rst
2022-04-19 10:13:08 +08:00

14 lines
319 B
ReStructuredText

colossalai.utils.memory\_tracer
===============================
.. automodule:: colossalai.gemini.memory_tracer
:members:
.. toctree::
:maxdepth: 2
colossalai.gemini.memory_tracer.async_memtracer
colossalai.gemini.memory_tracer.memstats_collector
colossalai.gemini.memory_tracer.model_data_memtracer