ColossalAI/docs/colossalai/colossalai.engine.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

13 lines
226 B
ReStructuredText

colossalai.engine
=================
.. automodule:: colossalai.engine
:members:
.. toctree::
:maxdepth: 2
colossalai.engine.gradient_accumulation
colossalai.engine.gradient_handler
colossalai.engine.schedule