RegressionRMSELoss

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

Bases: RegressionMSELoss

Root MSE loss between ground truth and predicted coordinates.

Methods Summary

compute_loss(targets, predictions)

Methods Documentation

compute_loss(targets: Tensor[Tensor], predictions: Tensor[Tensor]) → Tensor[Tensor][source]