HeatmapMSELoss

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

Bases: HeatmapLoss

MSE loss between heatmaps.

Methods Summary

compute_loss(targets, predictions)

Methods Documentation

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