A CLI and library for interacting with the Weights and Biases API.
Project description
Weights and Biases
The W&B client is an open source library and CLI (wandb) for organizing and analyzing your machine learning experiments. Think of it as a framework-agnostic lightweight TensorBoard that persists additional information such as the state of your code, system metrics, and configuration parameters.
Features
- Store config parameters used in a training run
- Associate version control with your training runs
- Search, compare, and visualize training runs
- Analyze system usage metrics alongside runs
- Collaborate with team members
- Run parameter sweeps
- Persist runs forever
Quickstart
pip install wandb
In your training script:
import wandb
# Your custom arguments defined here
args = ...
run = wandb.init(config=args)
run.config["more"] = "custom"
def training_loop():
while True:
# Do some machine learning
epoch, loss, val_loss = ...
# Framework agnostic / custom metrics
wandb.log({"epoch": epoch, "loss": loss, "val_loss": val_loss})
Running your script
Run wandb signup
from the directory of your training script. If you already have an account, you can run wandb init
to initialize a new directory. You can checkin wandb/settings to version control to share your project with other users.
Run your script with python my_script.py
and all metadata will be synced to the cloud. Data is staged locally in a directory named wandb relative to your script. If you want to test your script without syncing to the cloud you can run wandb off
.
Detailed Usage
Framework specific and detailed usage can be found in our documentation.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for wandb-0.6.26-py2.py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 4397d5c5b3af24f0e9d763eb234f2d929d306468f32be34f7173ff1a6f9e8b4e |
|
MD5 | 72dc1faca5ff1ac1c6743aa71a204ad1 |
|
BLAKE2b-256 | d846205ac1e002c2c05c0ce903e3b8cac46bb307114fe6bdd785b82dcf114bc2 |