ColossalAI/docs/colossalai/colossalai.utils.model.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
251 B
ReStructuredText

colossalai.utils.model
======================
.. automodule:: colossalai.utils.model
:members:
.. toctree::
:maxdepth: 2
colossalai.utils.model.colo_init_context
colossalai.utils.model.lazy_init_context
colossalai.utils.model.utils