HeatmapTracker
==============

.. currentmodule:: lightning_pose.models

.. autoclass:: HeatmapTracker
   :show-inheritance:

   .. rubric:: Methods Summary

   .. autosummary::

      ~HeatmapTracker.forward
      ~HeatmapTracker.get_loss_inputs_labeled
      ~HeatmapTracker.get_parameters
      ~HeatmapTracker.predict_step

   .. rubric:: Methods Documentation

   .. automethod:: forward
   .. automethod:: get_loss_inputs_labeled
   .. automethod:: get_parameters
   .. automethod:: predict_step