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

6 lines
113 B
ReStructuredText

colossalai.nn.loss.loss\_2d
===========================
.. automodule:: colossalai.nn.loss.loss_2d
:members: