mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-24 19:17:30 +00:00
@@ -8,6 +8,7 @@ colossalai.nn.loss
|
||||
.. toctree::
|
||||
:maxdepth: 2
|
||||
|
||||
colossalai.nn.loss.loss_1d
|
||||
colossalai.nn.loss.loss_2d
|
||||
colossalai.nn.loss.loss_2p5d
|
||||
colossalai.nn.loss.loss_3d
|
||||
|
Reference in New Issue
Block a user