Files
ColossalAI/docs/colossalai/colossalai.gemini.memory_tracer.rst
ver217 d068af81a3 [doc] update rst and docstring (#1351)
* update rst

* add zero docstr

* fix docstr

* remove fx.tracer.meta_patch

* fix docstr

* fix docstr

* update fx rst

* fix fx docstr

* remove useless rst
2022-07-21 15:54:53 +08:00

14 lines
320 B
ReStructuredText

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