ColossalAI/docs/colossalai/colossalai.nn.model.rst

colossalai.nn.model
===================

.. automodule:: colossalai.nn.model
   :members:

.. toctree::
   :maxdepth: 2

   colossalai.nn.model.vanilla_resnet
   colossalai.nn.model.vision_transformer


.. toctree::
   :maxdepth: 2

   colossalai.nn.model.base_model