以数据为中心的调试:面向训练神经网络的团队 [P]

Reddit r/MachineLearning 工具

摘要

WeightsLab 是一个开源、PyTorch 原生的工具,允许团队在训练过程中暂停、检查实时损失信号,并在数据问题(如标签错误和类别不平衡)影响模型性能之前发现它们。它专为处理图像、视频和 LiDAR 点云的计算机视觉工程师而设计。

我们刚刚对 WeightsLab 进行了重大改进,并想在此分享。如果你曾经花数小时调试训练过程,最终却发现一直是数据问题,那么这个工具就是为你准备的。WeightsLab 允许你在运行中暂停训练,检查实时损失信号,并在标签错误、类别不平衡和异常值破坏你的模型之前捕获它们。开源、PyTorch 原生,专为处理图像、视频和 LiDAR 点云数据的计算机视觉工程师打造。非常希望听到社区的看法,以及它是否看起来有用,并帮助更多人找到它:[ https://github.com/GrayboxTech/weightslab]
查看原文

相似文章

你存在于权重中吗?

Product Hunt

一个工具,让你检查你的数据是否被包含在大语言模型的训练集中,探索AI中的数字永生概念。

我们很高兴开源 LIDARLearn [R] [D] [P]

Reddit r/MachineLearning

LIDARLearn 是一个开源 PyTorch 库,用于 3D 点云深度学习,统一了 56 个预配置模型,内置交叉验证和自动生成出版级 LaTeX 报告的功能。该框架支持有监督学习、自监督学习和参数高效微调方法,适用于 ModelNet40、ShapeNet 和遥感基准等数据集。