HeatmapLoss
===========

.. currentmodule:: lightning_pose.losses.losses

.. autoclass:: HeatmapLoss
   :show-inheritance:

   .. rubric:: Methods Summary

   .. autosummary::

      ~HeatmapLoss.__call__
      ~HeatmapLoss.compute_loss
      ~HeatmapLoss.remove_nans

   .. rubric:: Methods Documentation

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