Skip to content

on_fit_epoch_end


Sends training metrics to Ray Tune at end of each epoch.

Source code in ultralytics/yolo/utils/callbacks/raytune.py
def on_fit_epoch_end(trainer):
    """Sends training metrics to Ray Tune at end of each epoch."""
    if ray.tune.is_session_enabled():
        metrics = trainer.metrics
        metrics['epoch'] = trainer.epoch
        session.report(metrics)




Created 2023-04-16, Updated 2023-05-17
Authors: Glenn Jocher (3)