What problem does it solve?
Provide a unified, collaborative platform to automatically track machine learning experiments, capture configuration and metrics, version datasets and models, and visualize training in real time so teams can reproduce results and ship models faster.
Core Features & Use Cases
- Experiment Tracking: Automatic logging of metrics, hyperparameters, system metrics, and training artifacts for each run.
- Real-time Visualization & Comparison: Live dashboards to compare runs, view custom charts, histograms, and media.
- Hyperparameter Sweeps: Integrated sweep configurations and agents for random, grid, or Bayesian optimization.
- Artifacts & Model Registry: Versioned dataset and model artifacts with lineage, aliases, and promotion workflows for staging/production.
- Framework Integrations: Examples and integrations for PyTorch, TensorFlow/Keras, PyTorch Lightning, HuggingFace, XGBoost, LightGBM, and fast.ai.
- Use Case: Run large-scale sweeps to tune an image classification pipeline, track dataset versions, and promote the best model to a registry for deployment.
Quick Start
Initialize a W&B run for your project, set key hyperparameters in the run config, log metrics and artifacts during training with wandb.log, and finish the run to upload results.