torchmetrics
TorchMetrics提供超过100种PyTorch指标的实现,拥有易用的API和支持自定义指标的功能,优化分布式训练并减少样板代码。它支持自动批次累积和设备同步,模块化设计使得指标可以自动放置在正确的设备上,并兼容PyTorch Lightning。覆盖音频、分类、检测、图像等多个领域,并提供可视化工具,便于理解机器学习算法。