Spaces:

AkashDataScience
/

FastSAM

Sleeping

FastSAM / ultralytics /yolo /utils /callbacks /raytune.py

Adding folder from ultralytics

00a2b0a over 2 years ago

495 Bytes

# Ultralytics YOLO 🚀, AGPL-3.0 license

try:
    import ray
    from ray import tune
    from ray.air import session
except (ImportError, AssertionError):
    tune = None


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)


callbacks = {
    'on_fit_epoch_end': on_fit_epoch_end, } if tune else {}

	# Ultralytics YOLO 🚀, AGPL-3.0 license

	try:
	import ray
	from ray import tune
	from ray.air import session
	except (ImportError, AssertionError):
	tune = None


	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)


	callbacks = {
	'on_fit_epoch_end': on_fit_epoch_end, } if tune else {}