BatchLoss
=========

.. currentmodule:: lib.training.loss

.. autoclass:: BatchLoss
   :show-inheritance:

   .. rubric:: Attributes Summary

   .. autosummary::

      ~BatchLoss.mask
      ~BatchLoss.total

   .. rubric:: Methods Summary

   .. autosummary::

      ~BatchLoss.to_cpu

   .. rubric:: Attributes Documentation

   .. autoattribute:: mask
   .. autoattribute:: total

   .. rubric:: Methods Documentation

   .. automethod:: to_cpu