How do I find an artifact from the best run in a sweep?

To retrieve artifacts from the best-performing run in a sweep, use the following code. It sorts the sweep’s runs by validation accuracy, selects the top run, and downloads each artifact that run logged.

api = wandb.Api()
sweep = api.sweep("entity/project/sweep_id")
runs = sorted(sweep.runs, key=lambda run: run.summary.get("val_acc", 0), reverse=True)
best_run = runs[0]
for artifact in best_run.logged_artifacts():
    artifact_path = artifact.download()
    print(artifact_path)

Each artifact downloads to a local path, which the script prints to the console.

Artifacts

⌘I