xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 21:44:22 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

ultralytics 8.1.10 MLFlow, OBB, TFLite and INT8 fixes (#8016)

Browse Source 
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Youho99 2024-02-07 02:28:14 +01:00 committed by GitHub
parent beada6e2be
commit da40839451
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ultralytics/__init__.py
View File

@ -1,6 +1,6 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics YOLO 🚀, AGPL-3.0 license
__version__ = "8.1.9" __version__ = "8.1.10"
from ultralytics.data.explorer.explorer import Explorer from ultralytics.data.explorer.explorer import Explorer
from ultralytics.models import RTDETR, SAM, YOLO from ultralytics.models import RTDETR, SAM, YOLO

8
ultralytics/utils/callbacks/mlflow.py
View File

@ -86,9 +86,12 @@ def on_train_epoch_end(trainer):
"""Log training metrics at the end of each train epoch to MLflow.""" """Log training metrics at the end of each train epoch to MLflow."""
if mlflow: if mlflow:
mlflow.log_metrics( mlflow.log_metrics(
metrics=SANITIZE(trainer.label_loss_items(trainer.tloss, prefix="train")), step=trainer.epoch metrics={
**SANITIZE(trainer.lr),
**SANITIZE(trainer.label_loss_items(trainer.tloss, prefix="train")),
},
step=trainer.epoch,
) )
mlflow.log_metrics(metrics=SANITIZE(trainer.lr), step=trainer.epoch)
def on_fit_epoch_end(trainer): def on_fit_epoch_end(trainer):
@ -115,6 +118,7 @@ def on_train_end(trainer):
callbacks = ( callbacks = (
{ {
"on_pretrain_routine_end": on_pretrain_routine_end, "on_pretrain_routine_end": on_pretrain_routine_end,
"on_train_epoch_end": on_train_epoch_end,
"on_fit_epoch_end": on_fit_epoch_end, "on_fit_epoch_end": on_fit_epoch_end,
"on_train_end": on_train_end, "on_train_end": on_train_end,
} }