clean_any_nans

lightning_pose.data.utils.clean_any_nans(data: Tensor, dim: int) → Tensor

Remove samples from a data array that contain nans.