HeatmapKLLoss
=============

.. currentmodule:: lightning_pose.losses.losses

.. autoclass:: HeatmapKLLoss
   :show-inheritance:

   .. rubric:: Attributes Summary

   .. autosummary::

      ~HeatmapKLLoss.loss_name

   .. rubric:: Methods Summary

   .. autosummary::

      ~HeatmapKLLoss.compute_loss

   .. rubric:: Attributes Documentation

   .. autoattribute:: loss_name

   .. rubric:: Methods Documentation

   .. automethod:: compute_loss