def loss(self,
             feats: Tuple[Tensor],
             batch_data_samples: OptSampleList,
             train_cfg: OptConfigType = {}) -> dict:
        """Calculate losses from a batch of inputs and data samples."""
Calculate losses from a batch of inputs and data samples.
mmpose/mmpose/models/heads/base_head.py