ColossalAI/docs/colossalai/colossalai.engine.rst at 75abc75c15f5c212c285808b718ee5cd60372068 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-04 17:58:00 +00:00

Files

Jiarui Fang 4165eabb1e [hotfix] remove potiential circle import (#1307 )

* make it faster

* [hotfix] remove circle import

2022-07-14 13:44:26 +08:00

13 lines

212 B

ReStructuredText

Raw Blame History

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