def METHOD_NAME(self, feats: Tuple[torch.Tensor],
         data_samples: List[ReIDDataSample]) -> dict:
    """Calculate losses.
    Args:
        feats (tuple[Tensor]): The features extracted from the backbone.
        data_samples (List[ReIDDataSample]): The annotation data of
            every samples.
    Returns:
        dict: a dictionary of loss components
    """
    # The part can be traced by torch.fx
    feats = self(feats)
    # The part can not be traced by torch.fx
    losses = self.loss_by_feat(feats, data_samples)
    return losses