Download and use artifacts - Weights & Biases Documentation

Download and use an artifact that is already stored on the W&B server or construct an artifact object and pass it in to for de-duplication as necessary.

Team members with a Models Viewer seat cannot download artifacts.

Download and use an artifact stored on W&B

Download an artifact as a tracked run input when it is a run dependency, or fetch it directly for workflows that don’t require lineage tracking.

References that have schemes that W&B knows how to handle get downloaded just like artifact files. For more information, see Track external files.

You can download all files in an artifact or download individual files:

Use Artifact.download() to download all files.
Use Artifact.get_entry() to retrieve a specific entry, and then call .download() to download it.

Track run input
Fetch directly
W&B CLI

Use wandb.Run.use_artifact() to download an artifact as part of a W&B run. W&B records the artifact as an input to the run.

Import the W&B Python SDK and initialize a W&B Run.
Retrieve the artifact version with wandb.Run.use_artifact().
Download all files or a specific file from the returned Artifact object.

import wandb    

with wandb.init(project="<example>", job_type="<job-type>") as run:
    # Indicate the artifact to use. Format is "name:alias"
    artifact = run.use_artifact("bike-dataset:latest")

    # Download the entire artifact
    datadir = artifact.download()

    # Download a specific file
    entry = artifact.get_entry("bike.png")
    file_path = entry.download() # downloads only bike.png

Use the Public API to download an artifact without creating a run or recording a run input relationship.

Import the W&B Python SDK and create a client object with wandb.Api().
Retrieve the artifact version with wandb.Api.artifact().
Download all files or a specific file from the returned Artifact object.

Specify the artifact path in the format entity/project/artifact:alias.

import wandb

api = wandb.Api()
artifact = api.artifact("entity/project/artifact:alias")
artifact.download()

Use the wandb artifact get command to download an artifact from the W&B server.

$ wandb artifact get project/artifact:alias --root mnist/

Partially download an artifact

You can optionally download part of an artifact based on a prefix. Use the path_prefix= (wandb.Artifact.download(path_prefix=)) parameter to download a single file or the content of a sub-folder.

with wandb.init(project="<example>", job_type="<job-type>") as run:
    # Indicate the artifact to use. Format is "name:alias"
    artifact = run.use_artifact("bike-dataset:latest")

    # Download a specific file or sub-folder
    artifact.download(path_prefix="bike.png") # downloads only bike.png

Alternatively, you can download files from a certain directory. To do so, specify the directory within the path_prefix= parameter. Continuing from the previous code snippet:

# downloads files in the images/bikes directory
artifact.download(path_prefix="images/bikes/") 

Use an artifact from a different project

Specify the name of artifact along with its project name to reference an artifact. You can also reference artifacts across entities by specifying the name of the artifact with its entity name. The following code example demonstrates how to query an artifact from another project as input to the current W&B run.

with wandb.init(project="<example>", job_type="<job-type>") as run:
    # Query W&B for an artifact from another project and mark it
    # as an input to this run.
    artifact = run.use_artifact("my-project/artifact:alias")

    # Use an artifact from another entity and mark it as an input
    # to this run.
    artifact = run.use_artifact("my-entity/my-project/artifact:alias")

Construct and use an artifact simultaneously

Simultaneously construct and use an artifact. Create an artifact object and pass it to use_artifact. This creates an artifact in W&B if it does not exist yet. The wandb.Run.use_artifact() API is idempotent, so you can call it as many times as you like.

import wandb

with wandb.init(project="<example>", job_type="<job-type>") as run:
    artifact = wandb.Artifact("reference model")
    artifact.add_file("model.h5")
    run.use_artifact(artifact)

For more information about constructing an artifact, see Construct an artifact.

​Download and use an artifact stored on W&B

​Partially download an artifact

​Use an artifact from a different project

​Construct and use an artifact simultaneously

Download and use an artifact stored on W&B

Partially download an artifact

Use an artifact from a different project

Construct and use an artifact simultaneously