weights-and-biases

Track ML experiments, run hyperparameter sweeps, and manage model artifacts with W&B.

13.0k|930|Updated Nov 3, 2025
One-click install
npx skills add https://github.com/Orchestra-Research/AI-research-SKILLs --skill weights-and-biases-orchestra-research
Or copy as Structured Prompt for Agent▼
Please help me install this Agent Skill.
Skill: weights-and-biases
Source: https://github.com/Orchestra-Research/AI-research-SKILLs/tree/main/13-mlops/weights-and-biases
Command: npx skills add https://github.com/Orchestra-Research/AI-research-SKILLs --skill weights-and-biases-orchestra-research

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill requires wandb, and includes references (resource) components.

What problem does it solve? Machine learning teams lose track of experiment configurations, metrics, and model versions when training runs are scattered across notebooks and scripts, making results hard to reproduce and compare. ## Core Features & Use Cases - Experiment Tracking: Automatically log metrics, configs, media, and system stats from PyTorch, TensorFlow, Keras, HuggingFace, and PyTorch Lightning training loops. - Hyperparameter Sweeps: Run grid, random, or Bayesian optimization searches with early termination and parallel agents across GPUs. - Artifacts & Model Registry: Version datasets and models with lineage tracking, aliases, and promotion workflows from staging to production. - Use Case: A team fine-tuning BERT models can log every run to a shared W&B project, launch a Bayesian sweep over learning rates and batch sizes, then promote the best checkpoint to a production model registry. ## Quick Start Initialize a W&B run in my training script, log the loss and accuracy each epoch, and save the final model as a versioned artifact.

Frequently Asked Questions about weights-and-biases

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I track PyTorch training experiments with Weights & Biases?▼

Call wandb.init with your project name and config, then use wandb.log inside your training loop to record metrics like loss and accuracy. Call wandb.finish at the end, and optionally save checkpoints with wandb.save or as artifacts.

How to run hyperparameter sweeps with W&B?▼

Define a sweep config with a search method (grid, random, or bayes), a target metric, and parameter distributions, then create it with wandb.sweep. Launch agents with wandb.agent pointing to a training function that reads hyperparameters from wandb.config.

Does W&B integrate with HuggingFace Transformers?▼

Yes, set report_to="wandb" in TrainingArguments and the Trainer automatically logs metrics, hyperparameters, and checkpoints to W&B. You can also add custom WandbCallback subclasses for additional logging.

Can I use W&B offline without an internet connection?▼

Yes, set the WANDB_MODE environment variable to "offline" before initializing your run. Metrics are stored locally and can be uploaded later using the wandb sync command on the run directory.

What is the difference between W&B artifacts and the model registry?▼

Artifacts are versioned files such as datasets or model checkpoints with automatic lineage tracking between runs. The model registry is a curated collection where model artifacts are linked and promoted through stages like staging and production using aliases.

Which sweep search method should I use in W&B?▼

Bayesian optimization is recommended for expensive training runs because it learns from previous trials to sample promising regions. Use grid search for few discrete parameters and random search for quick exploration across many parameters.