ColossalAI/docs/colossalai/colossalai.amp.rst

colossalai.amp
==============

.. automodule:: colossalai.amp
   :members:

.. toctree::
   :maxdepth: 2

   colossalai.amp.apex_amp
   colossalai.amp.naive_amp
   colossalai.amp.torch_amp


.. toctree::
   :maxdepth: 2

   colossalai.amp.amp_type
update examples and sphnix docs for the new api (#63) 2021-12-13 14:07:01 +00:00			`colossalai.amp`
removed tutorial markdown and refreshed rst files for consistency 2022-01-19 08:06:53 +00:00			`==============`

update rst (#615) 2022-04-01 07:46:38 +00:00			`.. automodule:: colossalai.amp`
			`:members:`
update examples and sphnix docs for the new api (#63) 2021-12-13 14:07:01 +00:00
			`.. toctree::`
			`:maxdepth: 2`

			`colossalai.amp.apex_amp`
			`colossalai.amp.naive_amp`
removed tutorial markdown and refreshed rst files for consistency 2022-01-19 08:06:53 +00:00			`colossalai.amp.torch_amp`
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.amp.amp_type`