def __init__(
        self,
        num_classes=1,
        mask_loss_weight=1.0,
        regr_loss_weight=1.0,
        size_average=True,
    ):
        """
        Args:
            num_classes (int): Number of classes in model.
                Default is ``1``.
            mask_loss_weight (float): heatmap loss weight coefficient.
                Default is ``1.0``.
            regr_loss_weight (float): HW regression loss weight coefficient.
                Default is ``1.0``.
            size_average (bool): loss batch scaling.
                Default is ``True``.
        """
        super().__init__()
        self.num_classes = num_classes
        self.mask_loss_weight = mask_loss_weight
        self.regr_loss_weight = regr_loss_weight
        self.size_average = size_average

        Args:
            num_classes (int): Number of classes in model.
                Default is ``1``.
            mask_loss_weight (float): heatmap loss weight coefficient.
                Default is ``1.0``.
            regr_loss_weight (float): HW regression loss weight coefficient.
                Default is ``1.0``.
            size_average (bool): loss batch scaling.
                Default is ``True``.
        
examples/detection/criterion.py