Quickstart

Documentation

Search…

English

Weights & Biases

Quickstart

Start tracking machine learning experiments in 5 minutes

Build better models more efficiently with Weights & Biases experiment tracking.
​Run a quick example project →​
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 the wandb library in a Python 3 environment. To login, you'll need to be signed in to you account at www.wandb.ai, then you will find your API key on the Authorize page.
Command Line
Notebook
1
pip install wandb
2
​
3
wandb login
Copied!
1
!pip install wandb
2
​
3
wandb.login()
Copied!
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() →​
1
import wandb
2
wandb.init(project="my-awesome-project")
Copied!
3. Track metrics
Use wandb.log() to track metrics or a framework integration for easy instrumentation.
More about wandb.log() →​
1
wandb.log({'accuracy': train_acc, 'loss': train_loss})
Copied!
4. Track hyperparameters
Save hyperparameters so you can quickly compare experiments.
More about wandb.config →​
1
wandb.config.dropout = 0.2
Copied!
5. Get alerts
Get notified via Slack or email if your W&B Run has crashed or whether a custom trigger, such as your loss going to NaN or a step in your ML pipeline has completed, has been reached. See the Alerts docs for a full setup.
​More about wandb.alert() →​
1.Turn on Alerts in your W&B User Settings​
2.Add wandb.alert() to your code
1
wandb.alert(
2
    title="Low accuracy", 
3
    text=f"Accuracy {acc} is below the acceptable threshold {thresh}"
4
)
Copied!
Then see W&B Alerts messages in Slack (or your email):
W&B Alerts in a Slack channel
What next?
1.​Collaborative Reports: Snapshot results, take notes, and share findings
2.​Data + Model Versioning: Track dependencies and results in your ML pipeline
3.​Data Visualization: Visualize and query datasets and model evaluations
4.​Hyperparameter Tuning: Quickly automate optimizing hyperparameters
5.​Private-Hosting: The enterprise solution for private cloud or on-prem hosting of W&B
Common Questions
Where do I find my API key?
Once you've signed in to www.wandb.ai, the API key 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 local, on-prem installs?
Yes, you can privately host W&B locally on your own machines or in a private cloud, try this quick tutorial notebook to see how. Note, to login to wandb local server you can set the host flag to the address of the local instance.  
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.