ultralytics/docs/en/reference/utils/callbacks/mlflow.md

description	keywords
Learn how to set up and customize MLflow logging for Ultralytics YOLO. Log metrics, parameters, and model artifacts easily.	MLflow, Ultralytics YOLO, logging, metrics, parameters, model artifacts, setup, tracking, customization

Reference for ultralytics/utils/callbacks/mlflow.py

!!! note

This file is available at [https://github.com/ultralytics/ultralytics/blob/main/ultralytics/utils/callbacks/mlflow.py](https://github.com/ultralytics/ultralytics/blob/main/ultralytics/utils/callbacks/mlflow.py). If you spot a problem please help fix it by [contributing](https://docs.ultralytics.com/help/contributing/) a [Pull Request](https://github.com/ultralytics/ultralytics/edit/main/ultralytics/utils/callbacks/mlflow.py) 🛠️. Thank you 🙏!

::: ultralytics.utils.callbacks.mlflow.sanitize_dict

---

::: ultralytics.utils.callbacks.mlflow.on_pretrain_routine_end

---

::: ultralytics.utils.callbacks.mlflow.on_train_epoch_end

---

::: ultralytics.utils.callbacks.mlflow.on_fit_epoch_end

---

::: ultralytics.utils.callbacks.mlflow.on_train_end