Start tracking machine learning experiments in 5 minutes
Build better models more efficiently with Weights & Biases experiment tracking.
Try this short Google Colab to see Weights & Biases in action, no code installation required!

1. Set up wandb

Sign up for a free account, then from the command line install our library in a Python 3 environment.
pip install wandb
wandb login
Trouble installing the library locally? Try this quick tutorial notebook.

2. Start a new run

Initialize a new run in W&B in your Python script or notebook. wandb.init() will start tracking system metrics and console logs, right out of the box. Run your code, put in your API key when prompted, and you'll see the new run appear in W&B. More about wandb.init() →
import wandb

3. Track metrics

Use wandb.log() to track metrics, or a framework integration for easy instrumentation. More about wandb.log() →
wandb.log({'accuracy': train_acc, 'loss': train_loss})

4. Track hyperparameters

Save hyperparameters so you can quickly compare experiments. More about wandb.config →
wandb.config.dropout = 0.2

What next?

  1. 1.
    Collaborative Reports: Snapshot results, take notes, and share findings
  2. 2.
    Data + Model Versioning: Track dependencies and results in your ML pipeline
  3. 3.
    Data Visualization: Visualize and query datasets and model evaluations
  4. 4.
    Hyperparameter Tuning: Quickly automate optimizing hyperparameters

Common Questions

Where do I find my API key? Once you've logged in, it will be on the Authorize page.
How do I use W&B in an automated environment? If you are training models in an automated environment where it's inconvenient to run shell commands, such as Google's CloudML, you should look at our guide to configuration with Environment Variables.
Do you offer on-prem installs? Yes, you can self-host W&B Local on your own machines or in a private cloud.
How do I turn off wandb logging temporarily? If you're testing code and want to disable wandb syncing, set the environment variable WANDB_MODE=offline.
Last modified 3mo ago