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