src/transformers/models/mask2former/modeling_mask2former.py/Mask2FormerForUniversalSegmentation/get_loss
class Mask2FormerForUniversalSegmentation:    def get_loss(self, loss_dict: Dict[str, Tensor]) -> Tensor:
        return sum(loss_dict.values())