PatchMasking

class lightning_pose.callbacks.PatchMasking

Bases: Callback

Callback to apply curriculum patch masking during training.

Methods Summary

on_train_batch_start(trainer, pl_module, ...)	Apply patch masking to the batch before it goes to the model.
on_train_epoch_end(trainer, pl_module)	Log curriculum progress.

Methods Documentation

on_train_batch_start(trainer, pl_module, batch, batch_idx)

Apply patch masking to the batch before it goes to the model.

on_train_epoch_end(trainer, pl_module) → None

Log curriculum progress.

__init__(patch_mask_config: dict | None = None, patch_seed: int = 0)

__new__(**kwargs)