misc/py-torchmetrics/pkg-descr


1
2
3
4
5
6
7
8

TorchMetrics is a collection of 90+ PyTorch metrics implementations and an
easy-to-use API to create custom metrics. It offers:
* A standardized interface to increase reproducibility
* Reduces Boilerplate
* Distributed-training compatible
* Rigorously tested
* Automatic accumulation over batches
* Automatic synchronization between multiple devices