ColossalAI/docs/colossalai/colossalai.nn.metric.rst

14 lines
235 B
ReStructuredText

colossalai.nn.metric
====================
.. automodule:: colossalai.nn.metric
:members:
.. toctree::
:maxdepth: 2
colossalai.nn.metric.accuracy_2d
colossalai.nn.metric.accuracy_2p5d
colossalai.nn.metric.accuracy_3d