ColossalAI/docs/colossalai/colossalai.nn.optimizer.rst

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

.. toctree::
   :maxdepth: 2

   colossalai.nn.optimizer.fused_adam
   colossalai.nn.optimizer.fused_lamb
   colossalai.nn.optimizer.fused_sgd
   colossalai.nn.optimizer.lamb
   colossalai.nn.optimizer.lars


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

			`.. toctree::`
			`:maxdepth: 2`

			`colossalai.nn.optimizer.fused_adam`
			`colossalai.nn.optimizer.fused_lamb`
			`colossalai.nn.optimizer.fused_sgd`
			`colossalai.nn.optimizer.lamb`
update documentation 2021-10-29 01:29:20 +00:00			`colossalai.nn.optimizer.lars`
update examples and sphnix docs for the new api (#63) 2021-12-13 14:07:01 +00:00

			`.. automodule:: colossalai.nn.optimizer`
			`:members:`