Update raytune.py

This commit is contained in:
JN03 2025-03-24 02:22:44 +07:00 committed by GitHub
parent 6fbaf42b23
commit b88c4b787a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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)