Reference for ultralytics/utils/callbacks/mlflow.py
Note
Full source code for this file is available at https://github.com/ultralytics/ultralytics/blob/main/ultralytics/utils/callbacks/mlflow.py. Help us fix any issues you see by submitting a Pull Request 🛠️. Thank you 🙏!
ultralytics.utils.callbacks.mlflow.on_pretrain_routine_end(trainer)
Logs training parameters to MLflow.
Source code in ultralytics/utils/callbacks/mlflow.py
ultralytics.utils.callbacks.mlflow.on_fit_epoch_end(trainer)
Logs training metrics to Mlflow.
ultralytics.utils.callbacks.mlflow.on_train_end(trainer)
Called at end of train loop to log model artifact info.