RegressionRMSELoss

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

Bases: RegressionMSELoss

Root MSE loss between ground truth and predicted coordinates.

Methods Summary

compute_loss(targets, predictions)

Methods Documentation

compute_loss(targets: Tensor, {'__torchtyping__': True, 'details': ('batch_x_two_x_num_keypoints',), 'cls_name': 'TensorType'}], predictions: Tensor, {'__torchtyping__': True, 'details': ('batch_x_two_x_num_keypoints',), 'cls_name': 'TensorType'}]) → Tensor, {'__torchtyping__': True, 'details': ('batch_x_num_keypoints',), 'cls_name': 'TensorType'}]

__init__(data_module: BaseDataModule | UnlabeledDataModule | None = None, epsilon: float = 0.0, log_weight: float = 0.0, **kwargs) → None

Parameters:

• data_module – give losses access to data for computing data-specific loss params

• epsilon – loss values below epsilon will be zeroed out

• log_weight – natural log of the weight in front of the loss term in the final objective function