Skip to navigation

aitune.utils.monitoring.hardware_metrics_annotation

View as Markdown

Annotation for collecting hardware metrics.

Module Contents

Classes

NameDescription
collect_hardware_metricsDecorator/context manager for collecting hardware metrics.

API

class aitune.utils.monitoring.hardware_metrics_annotation.collect_hardware_metrics(
name: str | None = None
)

Decorator/context manager for collecting hardware metrics.

aitune.utils.monitoring.hardware_metrics_annotation.collect_hardware_metrics.__call__(
func
)

Wraps the function as a decorator.

aitune.utils.monitoring.hardware_metrics_annotation.collect_hardware_metrics.__enter__()

Enters the context manager.

aitune.utils.monitoring.hardware_metrics_annotation.collect_hardware_metrics.__exit__(
exc_type,
exc_value,
traceback
)

Exits the context manager.