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

18 lines
267 B
ReStructuredText
Raw Normal View History

2021-10-28 16:21:23 +00:00
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