1a73a1367fedfa8368b6c42103e60e1b370bc14a,catalyst/contrib/criterion/focal_loss.py,FocalLoss,forward,#FocalLoss#Any#Any#,11

Before Change

                f"Got ({targets.size()}) and ({outputs.size()})"
            )

        max_val = (-outputs).clamp(min=0)
        log_ = ((-max_val).exp() + (-outputs - max_val).exp()).log()
        loss = outputs - outputs * targets + max_val + log_

        invprobs = F.logsigmoid(-outputs * (targets * 2.0 - 1.0))
        loss = (invprobs * self.gamma).exp() * loss

After Change

            if cls == self.ignore:
                continue

            cls_label_target = targets[..., cls].long()
            cls_label_input = logits[..., cls]

            loss += sigmoid_focal_loss(

Instances