telemetry.callbacks

telemetry.callbacks

Trainer callbacks for reporting runtime metrics at regular intervals.

Classes

Name	Description
TelemetryCallback	Trainer callback for tracking and reporting runtime metrics.

TelemetryCallback

telemetry.callbacks.TelemetryCallback()

Trainer callback for tracking and reporting runtime metrics.

This callback tracks training progress, runtime, and memory usage, sending telemetry at configurable intervals.

Methods

Name	Description
on_epoch_begin	Handle epoch start.
on_epoch_end	Handle epoch end.
on_step_end	Handle step end.
on_train_begin	Handle training start.
on_train_end	Handle training end.

on_epoch_begin

telemetry.callbacks.TelemetryCallback.on_epoch_begin(
    args,
    state,
    control,
    **kwargs,
)

Handle epoch start.

on_epoch_end

telemetry.callbacks.TelemetryCallback.on_epoch_end(
    args,
    state,
    control,
    **kwargs,
)

Handle epoch end.

on_step_end

telemetry.callbacks.TelemetryCallback.on_step_end(
    args,
    state,
    control,
    **kwargs,
)

Handle step end.

on_train_begin

telemetry.callbacks.TelemetryCallback.on_train_begin(
    args,
    state,
    control,
    **kwargs,
)

Handle training start.

on_train_end

telemetry.callbacks.TelemetryCallback.on_train_end(
    args,
    state,
    control,
    **kwargs,
)

Handle training end.