ColossalAI/docs/colossalai/colossalai.nn.rst

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

.. automodule:: colossalai.nn
   :members:

.. toctree::
   :maxdepth: 2

   colossalai.nn.layer
   colossalai.nn.loss
   colossalai.nn.lr_scheduler
   colossalai.nn.metric
   colossalai.nn.model
   colossalai.nn.optimizer


.. toctree::
   :maxdepth: 2

   colossalai.nn.init
Migrated project 2021-10-28 16:21:23 +00:00			`colossalai.nn`
			`=============`

update rst (#615) 2022-04-01 07:46:38 +00:00			`.. automodule:: colossalai.nn`
			`:members:`
removed tutorial markdown and refreshed rst files for consistency 2022-01-19 08:06:53 +00:00
Migrated project 2021-10-28 16:21:23 +00:00			`.. toctree::`
			`:maxdepth: 2`

			`colossalai.nn.layer`
			`colossalai.nn.loss`
			`colossalai.nn.lr_scheduler`
removed tutorial markdown and refreshed rst files for consistency 2022-01-19 08:06:53 +00:00			`colossalai.nn.metric`
Migrated project 2021-10-28 16:21:23 +00:00			`colossalai.nn.model`
			`colossalai.nn.optimizer`
update examples and sphnix docs for the new api (#63) 2021-12-13 14:07:01 +00:00

removed tutorial markdown and refreshed rst files for consistency 2022-01-19 08:06:53 +00:00			`.. toctree::`
			`:maxdepth: 2`

			`colossalai.nn.init`