5e46e590db971af21d1c71260ce885431718e2d3,inferno/extensions/metrics/categorical.py,CategoricalError,forward,#CategoricalError#Any#Any#,13

Before Change

        else:
            // Multiclass classificiation
            _, predicted_class = torch.max(prediction, 1)
            incorrect = predicted_class.squeeze(1).type_as(target).ne(target).float()
            if self.aggregation_mode == "mean":
                return incorrect.mean()
            else:
                return incorrect.sum()

After Change

        else:
            // Multiclass classificiation
            _, predicted_class = torch.max(prediction, 1)
            if predicted_class.dim() == prediction.dim():
                // Support for Pytorch 0.1.12
                predicted_class = predicted_class.squeeze(1)
            incorrect = predicted_class.type_as(target).ne(target).float()
            if self.aggregation_mode == "mean":
                return incorrect.mean()
            else:

Instances