EmpiricalEpsilon

class lightning_pose.losses.helpers.EmpiricalEpsilon(percentile: float)[source]

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: TensorType | ndarray) → float[source]

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

__init__(percentile: float) → None[source]