Passer au contenu principal
W&B registre est un dépôt central organisé de versions de W&B Artifact au sein de votre organisation. Les Users qui disposent des autorisations nécessaires dans votre organisation peuvent télécharger et utiliser des artifacts, les partager et gérer de manière collaborative le cycle de vie de tous les artifacts, quelle que soit l’équipe à laquelle l’utilisateur appartient. Utilisez le registre pour suivre les versions d’artifacts, auditer l’historique d’utilisation et des modifications d’un artifact, garantir la gouvernance et la conformité de vos artifacts, et automatiser les processus en aval, comme le CI/CD des modèles. En résumé, utilisez W&B registre pour : L’image suivante montre la page d’accueil de W&B registre. Un registre nommé Model est marqué d’une étoile. Deux collections sont affichées : DemoModels et Zoo_Classifier_Models.
W&B Registry

Découvrez les bases

Chaque organisation contient initialement deux registres que vous pouvez utiliser pour organiser vos artifacts de modèles et de jeux de données, appelés respectivement Models et Datasets. Vous pouvez créer des registres supplémentaires pour organiser d’autres types d’artifact selon les besoins de votre organisation. Chaque registre se compose d’une ou plusieurs collections. Chaque collection correspond à une tâche ou à un cas d’utilisation distinct. Pour ajouter un artifact à un registre, vous devez d’abord journaliser une version spécifique d’artifact dans W&B. Chaque fois que vous journalisez un artifact, W&B lui attribue automatiquement une version. Les versions d’artifact utilisent une indexation à partir de 0 : la première version est donc v0, la deuxième v1, et ainsi de suite. Une fois qu’un artifact est journalisé dans W&B, vous pouvez ensuite lier cette version spécifique d’artifact à une collection dans le registre.
Le terme “lien” désigne des pointeurs qui relient l’emplacement où W&B stocke l’artifact et l’emplacement où l’artifact est accessible dans le registre. W&B ne duplique pas les artifacts lorsque vous liez un artifact à une collection.
Par exemple, l’exemple de code suivant journalise et lie un artifact de modèle appelé "my_model.txt" à une collection nommée "first-collection" dans un registre appelé "model" :
  1. Initialisez un run W&B avec wandb.init().
  2. Journalisez l’artifact dans W&B avec wandb.Run.log().
  3. Indiquez le nom de la collection et du registre auxquels lier votre version d’artifact.
  4. Liez l’artifact à la collection à l’aide de wandb.Run.link_artifact().
Enregistrez ce code Python dans un script, puis exécutez-le. La version 0.18.6 ou ultérieure du SDK Python W&B est requise.
hello_collection.py
import wandb
import random

# Initialiser un run W&B pour suivre l'artifact
with wandb.init(project="registry_quickstart") as run:
    # Créer un fichier de modèle simulé pour pouvoir l'enregistrer
    with open("my_model.txt", "w") as f:
        f.write("Model: " + str(random.random()))

    # Enregistrer l'artifact dans W&B
    logged_artifact = run.log_artifact(
        artifact_or_path="./my_model.txt", 
        name="gemma-finetuned", 
        type="model" # Spécifie le type d'artifact
    )

    # Spécifier le nom de la collection et du registre
    # dans lesquels vous souhaitez publier l'artifact
    COLLECTION_NAME = "first-collection"
    REGISTRY_NAME = "model"

    # Lier l'artifact au registre
    run.link_artifact(
        artifact=logged_artifact, 
        target_path=f"wandb-registry-{REGISTRY_NAME}/{COLLECTION_NAME}"
    )
W&B crée automatiquement une collection pour vous si la collection que vous spécifiez dans la méthode wandb.Run.link_artifact(target_path = "") de l’objet run renvoyé n’existe pas dans le registre que vous indiquez. En reprenant l’exemple précédent, après avoir exécuté le script, accédez à W&B registre pour voir les versions d’artifact que vous et les autres membres de votre organisation publiez. Sélectionnez registre dans la barre latérale du projet, sous Applications. Sélectionnez le registre "Model". Dans le registre, vous devriez voir la collection "first-collection" avec votre version d’artifact liée. Une fois qu’une version d’artifact est liée à une collection dans un registre, les membres de votre organisation peuvent voir, télécharger, organiser et gérer vos versions d’artifact, créer des automatisations en aval, et plus encore, s’ils disposent des autorisations appropriées.
Si une version d’artifact enregistre des métriques (par exemple avec wandb.Run.log_artifact()), vous pouvez consulter les métriques de cette version depuis sa page de détails, et comparer les métriques entre plusieurs versions d’artifact depuis la page de la collection. Référez-vous à Voir les artifacts liés dans un registre.

Activer W&B Registry

Selon votre type de déploiement, remplissez les conditions suivantes pour activer W&B Registry :
Type de déploiementComment l’activer
Cloud mutualiséAucune action requise. W&B Registry est disponible dans W&B App.
Cloud dédiéContactez l’équipe en charge de votre compte pour activer W&B Registry pour votre déploiement.
AutogéréPour Server v0.70.0 ou version ultérieure, aucune action requise. Pour les anciennes versions de Server prises en charge, définissez la variable d’environnement ENABLE_REGISTRY_UI sur true. Référez-vous à Configurer les variables d’environnement.

Ressources pour bien démarrer

Selon votre cas d’usage, consultez les ressources suivantes pour démarrer avec W&B Registry :
  • Regardez la vidéo de tutoriel :
  • Suivez le cours W&B Model CI/CD et découvrez comment :
    • Utiliser W&B Registry pour gérer et versionner vos artifacts, suivre la traçabilité et faire passer vos modèles par différentes étapes de leur cycle de vie.
    • Automatiser vos flux de travail de gestion des modèles à l’aide de webhooks.
    • Intégrer W&B Registry à des systèmes et outils ML externes pour l’évaluation, la surveillance et le déploiement des modèles.