PatchMasking
============

.. currentmodule:: lightning_pose.callbacks

.. autoclass:: PatchMasking
   :show-inheritance:

   .. rubric:: Methods Summary

   .. autosummary::

      ~PatchMasking.on_train_batch_start
      ~PatchMasking.on_train_epoch_end

   .. rubric:: Methods Documentation

   .. automethod:: on_train_batch_start
   .. automethod:: on_train_epoch_end