From 318a218cbdfcb5e751f66c9a916207a67eee2a2b Mon Sep 17 00:00:00 2001 From: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com> Date: Mon, 11 Nov 2024 00:03:59 +0800 Subject: [PATCH] Fix Ray Tune error (#17416) Co-authored-by: UltralyticsAssistant Co-authored-by: Glenn Jocher --- ultralytics/utils/callbacks/raytune.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/ultralytics/utils/callbacks/raytune.py b/ultralytics/utils/callbacks/raytune.py index 1a368db6..d92dc322 100644 --- a/ultralytics/utils/callbacks/raytune.py +++ b/ultralytics/utils/callbacks/raytune.py @@ -16,8 +16,7 @@ def on_fit_epoch_end(trainer): """Sends training metrics to Ray Tune at end of each epoch.""" if ray.train._internal.session._get_session(): # replacement for deprecated ray.tune.is_session_enabled() metrics = trainer.metrics - metrics["epoch"] = trainer.epoch - session.report(metrics) + session.report({**metrics, **{"epoch": trainer.epoch + 1}}) callbacks = (