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

Merge b88c4b787aa67e63bae5ffb01d0f63ba5aadbd32 into 453c6e38a51e9d1d5a2aa5fb7f1014a711913397

Browse Source
This commit is contained in:
JN03 2025-03-23 19:27:21 +00:00 committed by GitHub
commit 40b760fbe5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ultralytics/utils/callbacks/raytune.py
View File

@ -14,7 +14,7 @@ except (ImportError, AssertionError):
def on_fit_epoch_end(trainer): def on_fit_epoch_end(trainer):
"""Sends training metrics to Ray Tune at end of each epoch.""" """Sends training metrics to Ray Tune at end of each epoch."""
if ray.tune.is_session_enabled(): if ray.train._internal.session.get_session():
metrics = trainer.metrics metrics = trainer.metrics
metrics["epoch"] = trainer.epoch metrics["epoch"] = trainer.epoch
session.report(metrics) session.report(metrics)