What problem does it solve?
Weights & Biases simplifies the process of managing ML experiments, allowing for real-time tracking, model registry, and collaborative ML projects.
Core Features & Use Cases
- Experiment Tracking: Automatically log metrics, visualize training, compare runs, and optimize hyperparameters.
- Model Registry: Manage model versions, lineages, and versioning for reproducibility.
- Collaboration: Share runs with team members and collaborate on ML projects.
- Use Case: When you're working on a machine learning project, W&B helps you keep track of all your experiments, compare the performance of different models, and easily share your work with your team.
Quick Start
To start using Weights & Biases, first install it with pip install wandb. Then, log in with wandb login or set your API key programmatically. Here's an example of logging an experiment:
import wandb
run = wandb.init(project="my-project")
for epoch in range(10):
# Your training code here
train_loss = train_epoch()
val_loss = validate()
wandb.log({
"epoch": epoch,
"train/loss": train_loss,
"val/loss": val_loss
})
wandb.finish()