How to use TorchMetrics

A guide for using TorchMetrics with PyTorch or PyTorch Lightning

Mattia Gatti
Towards AI
Published in
5 min readAug 30, 2022

--

Image by Thomas Wolter from Pixabay

A critical part of good training is to properly manage validation metrics. This is required in order to monitor which state of the model is performing the best. Different tasks require different metrics to evaluate the accuracy of the model and implementing them generally requires writing boilerplate code…

--

--

Currently doing research into AI Remote Sensing. Writing about Deep Learning and Geospatial Data Analysis.