Model Management

Search…

Model Management

Manage the model lifecycle from training to production

Use the W&B Model Registry as a central system of record for models.
Create Registered Models to organize your best model versions for a given task
Track a model moving into staging and production
See a history of all changes, including who moved a model to production
Model Registry Quickstart
1.Open your Model Registry and create a registered model.
2.In your script, log a model as an artifact version.
​​
​​
art = wandb.Artifact("my-object-detector", type="model")
art.add_file("saved_model_weights.pt")
wandb.log_artifact(art)
3.From the Artifact page, link the artifact version to the registry.
Watch the 1 minute video walkthrough
Model Registry Features
Model Versioning
Iterate to get the best model version for a task, and catalog all the changes along the way.
Track every model version in a central repository
Browse and compare model versions
Capture training metrics and hyperparameters
Model Lineage
Document and reproduce the complete pipeline of model training and evaluation.
Identify the exact dataset version the model trained on
Restore the training code, including git commit and diff patch
Get back to the model’s hyperparameters and other metadata for reproducibility
Dig in to upstream jobs that can affect model performance
Model Lifecycle
Manage the process as a model moves from training through staging to production.
Highlight the best model versions that are being evaluated for production
Communicate where a model version is in the process — staging, production etc
Review the history of model versions that moved through each stage
Model Registry Pilot Limits
This new feature is now turned on for all users to try for free, up to:
5 Registered Models, with unlimited versions linked to each model
10 most recent steps of Action History shown in the UI for each registered model
Up Next
Dig into the details of how to use Weights & Biases for model management:
Model Management Concepts
Model Management Walkthrough

Artifacts Examples

Model Management Concepts

Last modified 2d ago

Copy link

Outline

Model Registry Quickstart

Watch the 1 minute video walkthrough

Model Registry Features

Model Versioning

Model Lineage

Model Lifecycle

Model Registry Pilot Limits

Up Next