EmpiricalEpsilon

class lightning_pose.losses.helpers.EmpiricalEpsilon(percentile: float)

Bases: object

Find percentile value of a given loss tensor.

Methods Summary

__call__(loss)

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

Methods Documentation

__call__(loss: Tensor | array) → float

Compute the percentile of some loss, to use an for 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