RegressionTracker
=================

.. currentmodule:: lightning_pose.models.regression_tracker

.. autoclass:: RegressionTracker
   :show-inheritance:

   .. rubric:: Methods Summary

   .. autosummary::

      ~RegressionTracker.forward
      ~RegressionTracker.get_loss_inputs_labeled
      ~RegressionTracker.get_parameters
      ~RegressionTracker.predict_step

   .. rubric:: Methods Documentation

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