ColossalAI/docs/colossalai/colossalai.nn.rst

22 lines
306 B
ReStructuredText
Raw Normal View History

2021-10-28 16:21:23 +00:00
colossalai.nn
=============
.. automodule:: colossalai.nn
:members:
2021-10-28 16:21:23 +00:00
.. toctree::
:maxdepth: 2
colossalai.nn.layer
colossalai.nn.loss
colossalai.nn.lr_scheduler
colossalai.nn.metric
2021-10-28 16:21:23 +00:00
colossalai.nn.model
colossalai.nn.optimizer
.. toctree::
:maxdepth: 2
colossalai.nn.init