Welcome to TorchMetrics

TorchMetrics is a collection of 100+ PyTorch metrics implementations and an easy-to-use API to create custom metrics. It offers:

You can use TorchMetrics in any PyTorch model, or within PyTorch Lightning to enjoy the following additional benefits:

Your data will always be placed on the same device as your metrics
You can log Metric objects directly in Lightning to reduce even more boilerplate

Install TorchMetrics

For pip users

pip install torchmetrics

Or directly from conda

conda install -c conda-forge torchmetrics

Use this quickstart guide to learn key concepts.

Easily use TorchMetrics in your PyTorch Lightning code.

View the full list of metrics and filter by task and data type.

A detailed overview of the TorchMetrics API and concepts.

Learn how to implement a custom metric with TorchMetrics.

Detailed descriptions of each API package.