EmpiricalEpsilon

class lightning_pose.utils.pca.EmpiricalEpsilon

Bases: object

Find percentile value of a given loss tensor.

Methods Summary

__call__(loss)

Compute the percentile of some loss, to use as an epsilon-insensitive loss.

Methods Documentation

__call__(loss: Tensor | ndarray) → float

Compute the percentile of some loss, to use as an epsilon-insensitive loss.

Parameters:

loss – tensor with scalar loss per term (e.g., loss per image, or loss per keypoint, etc.)

Returns:

the percentile of the loss which we use as epsilon

__init__(percentile: float) → None

Initialize EmpiricalEpsilon.

Parameters:

percentile – the percentile (0–100) of the loss distribution to use as epsilon.

__new__(**kwargs)