From b88c4b787aa67e63bae5ffb01d0f63ba5aadbd32 Mon Sep 17 00:00:00 2001
From: JN03 <112270477+JNX03@users.noreply.github.com>
Date: Mon, 24 Mar 2025 02:22:44 +0700
Subject: [PATCH] Update raytune.py
---
ultralytics/utils/callbacks/raytune.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/ultralytics/utils/callbacks/raytune.py b/ultralytics/utils/callbacks/raytune.py
index f2694554..9779f9a1 100644
--- a/ultralytics/utils/callbacks/raytune.py
+++ b/ultralytics/utils/callbacks/raytune.py
@@ -14,7 +14,7 @@ except (ImportError, AssertionError):
def on_fit_epoch_end(trainer):
"""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["epoch"] = trainer.epoch
session.report(metrics)