Перейти к содержимому

Ссылка для ultralytics/utils/callbacks/raytune.py

Примечание

Этот файл доступен по адресу https://github.com/ultralytics/ ultralytics/blob/main/ ultralytics/utils/callbacks/raytune .py. Если ты заметил проблему, пожалуйста, помоги исправить ее, отправив Pull Request 🛠️. Спасибо 🙏!



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

В конце каждой эпохи отправляй метрики тренировки в Ray Tune.

Исходный код в ultralytics/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)





Создано 2023-11-12, Обновлено 2023-11-25
Авторы: glenn-jocher (3)