Weights & Biases Downloader Tool

Weights & Biases Downloader Tool#

The Weights & Biases (W&B) Downloader Tool helps download run logs from W&B to local JSON files.

Description#

This tool allows you to download run logs and metadata from Weights & Biases projects. It saves run data to JSON files in a structured directory format, making it easier to process W&B experiment data offline or archive results.

Usage#

ml_research_tools wandb-downloader [options]
Usage: ml_research_tools wandb-downloader [-h] [--entity ENTITY]
                                          [--project PROJECT]
                                          [--output-dir OUTPUT_DIR]
                                          [--timeout TIMEOUT] [--quiet]
                                          [--no-delete]

Download artifacts and runs from Weights & Biases

Options:
  -h, --help            show this help message and exit
  --entity, -e ENTITY   W&B entity (username or team name). Can also use
                        WANDB_ENTITY env variable.
  --project, -p PROJECT
                        W&B project name. Can also use WANDB_PROJECT env
                        variable.
  --output-dir, -o OUTPUT_DIR
                        Directory to save log files (default: wandb_logs)
  --timeout, -t TIMEOUT
                        API timeout in seconds (default: 30)
  --quiet, -q           Suppress progress bar and detailed logging
  --no-delete           Don't delete logs for runs that no longer exist

Examples#

Download all runs from a project:

ml_research_tools wandb-downloader --entity myuser --project myproject

Custom output directory:

ml_research_tools wandb-downloader -e myuser -p myproject -o ./experiment_logs

Using environment variables:

export WANDB_ENTITY=myuser
export WANDB_PROJECT=myproject
ml_research_tools wandb-downloader