tlc.client.torch.metrics.metrics_collectors.metrics_collector_base#

Base class for all metrics collectors in PyTorch.

Module Contents#

Classes#

Class

Description

MetricsCollector

API#

class tlc.client.torch.metrics.metrics_collectors.metrics_collector_base.MetricsCollector(compute_aggregates: bool = True)#
reset() None#
abstract compute_metrics(batch: tlc.core.builtins.types.SampleData, predictions: tlc.core.builtins.types.SampleData | None | torch.Tensor, hook_outputs: dict[int, torch.Tensor] | None = None) dict[str, tlc.core.builtins.types.MetricData]#
property column_schemas: dict[str, tlc.core.schema.Schema]#
property model: torch.nn.Module | None#
property layers: list[int]#
property aggregate_values: dict[str, float]#