コンテンツへスキップ

参考 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)