e839626c28d8dc6030ce71f3619e934b36fb6627,catalyst/metrics/functional.py,,process_multiclass_components,#,33

Before Change

        )

    if targets.max() != len(torch.unique(targets)) - 1:
        if raise_class_labels_mismatch:
            raise Exception(
                "`targets` maximum does not represent number of classes"
            )
        // mapping classes
        outputs, targets = map_labels_to_classes(outputs, targets)
    return outputs, targets


def process_multilabel_components(

After Change

        // we could find number of classes as max available number
        num_classes = max(
            int(outputs.max().detach().item() + 1),
            int(targets.max().detach().item() + 1),
        )

    if outputs.dim() == 1:

Instances