Loss

class lightning_pose.losses.losses.Loss(data_module: BaseDataModule | UnlabeledDataModule | None = None, epsilon: float | List[float] = 0.0, log_weight: float = 0.0, **kwargs)

Bases: object

Parent class for all losses.

Attributes Summary

weight

Methods Summary

__call__(*args, **kwargs)	Call self as a function.
compute_loss(**kwargs)
log_loss(loss, stage)
rectify_epsilon(loss)
reduce_loss(loss[, method])
remove_nans(**kwargs)

Attributes Documentation

weight

Methods Documentation

__call__(*args, **kwargs)

Call self as a function.

compute_loss(**kwargs)

log_loss(loss: Tensor, stage: Literal['train', 'val', 'test']) → List[dict]

rectify_epsilon(loss: Tensor) → Tensor

reduce_loss(loss: Tensor, method: str = 'mean') → Tensor[Tensor]

remove_nans(**kwargs)