Skip to main content

A CLI and library for interacting with the Weights and Biases API.

Project description

<div align="center">
<img src="" width="350" /><br><br>

# Weights and Biases [![ci](]( [![pypi](](

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` 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`.

<p align="center">
<img src="" alt="Runs screenshot" style="max-width:100%;">

## Detailed Usage

Framework specific and detailed usage can be found in our [documentation](

Project details

Release history Release notifications

This version
History Node


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Filename, size & hash SHA256 hash help File type Python version Upload date
wandb-0.6.6-py2.py3-none-any.whl (80.4 kB) Copy SHA256 hash SHA256 Wheel py2.py3
wandb-0.6.6.tar.gz (294.5 kB) Copy SHA256 hash SHA256 Source None

Supported by

Elastic Elastic Search Pingdom Pingdom Monitoring Google Google BigQuery Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN SignalFx SignalFx Supporter DigiCert DigiCert EV certificate StatusPage StatusPage Status page