How to log IntersectionOverUnion with mlflow logger and lightning? I use lightning IntersectionOverUnion with the class_metrics=True
