torchmetrics - torchmetrics技术,学习,经验文章

fengbingchun

5 小时前

深度学习中评估指标计算库TorchMetrics的使用TorchMetrics是一个包含100多个PyTorch指标实现的集合(如分类、检测、分割、回归等)，并提供易于使用的API来创建自定义指标。可以将TorchMetrics与任何PyTorch模型或PyTorch Lightning结合使用。源码地址：https://github.com/Lightning-AI/torchmetrics，最新发布版本为v1.9.0，license为Apache-2.0。