Skip to content

Reference for ultralytics/utils/callbacks/raytune.py

Note

Full source code for this file is available at https://github.com/ultralytics/ultralytics/blob/main/ultralytics/utils/callbacks/raytune.py. Help us fix any issues you see by submitting a Pull Request 🛠️. Thank you 🙏!

ultralytics.utils.callbacks.raytune.on_fit_epoch_end(trainer)

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

Source code in ultralytics/utils/callbacks/raytune.py 
15
16
17
18
19
20
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-07-16, Updated 2023-08-07
Authors: glenn-jocher (5), Laughing-q (1)