热门
导航
快讯
推荐文章
热门
导航
快讯
推荐文章
#TorchMetrics
torchmetrics - 100+ PyTorch 指标实现的集合,以及一个易于使用的 API 来创建自定义指标
Github
开源项目
PyTorch
分布式训练
机器学习
TorchMetrics
度量
TorchMetrics提供超过100种PyTorch指标的实现,拥有易用的API和支持自定义指标的功能,优化分布式训练并减少样板代码。它支持自动批次累积和设备同步,模块化设计使得指标可以自动放置在正确的设备上,并兼容PyTorch Lightning。覆盖音频、分类、检测、图像等多个领域,并提供可视化工具,便于理解机器学习算法。
1
1
相关文章
TorchMetrics: 机器学习中强大而灵活的指标计算工具
3 个月前
使用协议
隐私政策
广告服务
投诉举报邮箱: service@vectorlightyear.com
@2024 懂AI
·
鲁ICP备2024100362号-6
·
鲁公网安备37021002001498号