AnnealWeight

class lightning_pose.callbacks.AnnealWeight(attr_name: str, init_val: float = 0.0, increase_factor: float = 0.01, final_val: float = 1.0, freeze_until_epoch: int = 0)

Bases: Callback

Callback to change weight value during training.

Methods Summary

on_train_epoch_start(trainer, pl_module)	Called when the train epoch begins.
on_train_start(trainer, pl_module)	Called when the train begins.

Methods Documentation

on_train_epoch_start(trainer: Trainer, pl_module: LightningModule) → None

Called when the train epoch begins.

on_train_start(trainer, pl_module) → None

Called when the train begins.