d9172919d00695f0dddb0ab10c89c3b7a8e26427,mmdet/models/losses/ghm_loss.py,GHMR,__init__,#GHMR#Any#Any#Any#Any#,110

Before Change

        super(GHMR, self).__init__()
        self.mu = mu
        self.bins = bins
        self.edges = torch.arange(bins + 1).float().cuda() / bins
        self.edges[-1] = 1e3
        self.momentum = momentum
        if momentum > 0:
            self.acc_sum = torch.zeros(bins).cuda()
        self.loss_weight = loss_weight

    // TODO: support reduction parameter

After Change

        self.momentum = momentum
        if momentum > 0:
            acc_sum = torch.zeros(bins)
            self.register_buffer("acc_sum", acc_sum)
        self.loss_weight = loss_weight

    // TODO: support reduction parameter
    def forward(self, pred, target, label_weight, avg_factor=None):

Instances