tlc.client.torch.metrics.metrics_collectors#

Implementations of metrics collectors in PyTorch.

Modules and Packages#

Module

Description

metrics_collector_base

Base class for all metrics collectors in PyTorch.

embeddings_metrics_collector

Collect embeddings from PyTorch models.

classification_metrics_collector

Collect loss and predictions for classification problems.

functional_metrics_collector

Collect metrics using a custom function.

bounding_box_metrics_collector

Collect metrics for bounding box predictions.

segmentation_metrics_collector

Collect metrics for segmentation problems.