ColossalAI/docs/colossalai/colossalai.nn.loss.rst

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

.. toctree::
   :maxdepth: 2

   colossalai.nn.loss.cross_entropy_2d
   colossalai.nn.loss.cross_entropy_2p5d
   colossalai.nn.loss.cross_entropy_3d


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

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

			`colossalai.nn.loss.cross_entropy_2d`
			`colossalai.nn.loss.cross_entropy_2p5d`
			`colossalai.nn.loss.cross_entropy_3d`
update examples and sphnix docs for the new api (#63) 2021-12-13 14:07:01 +00:00

			`.. automodule:: colossalai.nn.loss`
			`:members:`