weights-and-biases

Track ML experiments, hyperparameter sweeps, and model artifacts with Weights & Biases.

Updated Jul 10, 2026
One-click install
npx skills add https://github.com/AvaTar-ArTs/.Agent-skills --skill weights-and-biases-avatar-arts
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: weights-and-biases
Source: https://github.com/AvaTar-ArTs/.Agent-skills/tree/main/skills/mlops/evaluation/weights-and-biases
Command: npx skills add https://github.com/AvaTar-ArTs/.Agent-skills --skill weights-and-biases-avatar-arts

SYSTEM DOCUMENTATION & REQUIREMENTS

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

What problem does it solve? Machine learning teams lose track of experiments, hyperparameters, and model versions when training runs are scattered across notebooks and scripts. This Skill provides complete guidance for instrumenting training code with Weights & Biases so every run, metric, and artifact is logged, comparable, and reproducible. ## Core Features & Use Cases - Experiment Tracking: 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 data scientist fine-tuning a BERT classifier can launch a Bayesian sweep over learning rate and batch size, compare 50 runs in a real-time dashboard, and promote the best checkpoint to a production model registry. ## Quick Start Ask the AI to instrument your PyTorch training script with W&B experiment tracking, logging loss and accuracy per epoch and saving 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 experiments with Weights & Biases?

Call wandb.init with your project name and config, then call wandb.log with metrics like loss and accuracy inside your training loop. Finish with wandb.finish, and optionally save checkpoints with wandb.save or as versioned artifacts.

How to run hyperparameter sweeps with wandb?

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 one or more agents with wandb.agent pointing to your training function to run trials.

Does W&B integrate with HuggingFace Transformers?

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

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 layer where model artifacts are linked and promoted through stages like staging and production using aliases.

Can I use wandb without an internet connection?

Yes, set WANDB_MODE=offline before initializing your run and all metrics are stored locally. Later, run wandb sync on the run directory to upload the logged data to the W&B servers.

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

Use Bayesian optimization for expensive training runs with limited compute, since it learns from previous trials. Use random search for quick exploration of many parameters, and grid search only for a few discrete parameters needing exhaustive coverage.