clean_any_nans

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

Remove samples from a data array that contain nans.