UnimodalLoss
============

.. currentmodule:: lightning_pose.losses.losses

.. autoclass:: UnimodalLoss
   :show-inheritance:

   .. rubric:: Attributes Summary

   .. autosummary::

      ~UnimodalLoss.LOSS_NAME_JS
      ~UnimodalLoss.LOSS_NAME_KL
      ~UnimodalLoss.LOSS_NAME_MSE

   .. rubric:: Methods Summary

   .. autosummary::

      ~UnimodalLoss.__call__
      ~UnimodalLoss.compute_loss
      ~UnimodalLoss.remove_nans

   .. rubric:: Attributes Documentation

   .. autoattribute:: LOSS_NAME_JS
   .. autoattribute:: LOSS_NAME_KL
   .. autoattribute:: LOSS_NAME_MSE

   .. rubric:: Methods Documentation

   .. automethod:: __call__
   .. automethod:: compute_loss
   .. automethod:: remove_nans