mirror of
https://github.com/THU-MIG/yolov10.git
synced 2025-05-23 05:24:22 +08:00
5065ca36a8
Co-authored-by: JF Chen <k-2feng@hotmail.com> Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
18 lines
385 B
Python
18 lines
385 B
Python
try:
|
|
import ray
|
|
from ray import tune
|
|
from ray.air import session
|
|
except (ImportError, AssertionError):
|
|
tune = None
|
|
|
|
|
|
def on_fit_epoch_end(trainer):
|
|
if ray.tune.is_session_enabled():
|
|
metrics = trainer.metrics
|
|
metrics['epoch'] = trainer.epoch
|
|
session.report(metrics)
|
|
|
|
|
|
callbacks = {
|
|
'on_fit_epoch_end': on_fit_epoch_end, } if tune else {}
|