Data-centric debugging for teams training neural nets [P]

Reddit r/MachineLearning Tools

Summary

WeightsLab is an open-source, PyTorch-native tool that allows teams to pause training, inspect live loss signals, and catch data issues like mislabels and class imbalance before they affect model performance. It is designed for computer vision engineers working with images, videos, and LiDAR point clouds.

We just did a big revamp of WeightsLab and wanted to share it here. If you’ve ever spent hours debugging a training run only to discover it was a data problem all along, this is for you. WeightsLab lets you pause training mid-run, inspect your live loss signals, and catch mislabels, class imbalance & outliers before they tank your model. Open source, PyTorch-native, built for CV engineers working with images, videos & LiDAR point cloud data. Would love to hear what the community thinks and if it looks useful, and helps more people find it: [ https://github.com/GrayboxTech/weightslab]
Original Article

Similar Articles

WARP: Weight-Space Analysis for Recovering Training Data Portfolios

arXiv cs.LG

WARP is a framework that recovers the domain mixture weights of a fine-tuned model from its released weights by generating pseudo-checkpoints via model merging and extracting geometric features. It achieves low mean absolute error on BERT and GPT-2, outperforming membership inference.

Are you in the Weights?

Product Hunt

A tool that lets you check if your data is included in the training sets of large language models, exploring the concept of digital immortality in AI.