Journaliser des modèles

Ce guide explique comment journaliser des modèles dans un run W&B et comment interagir avec eux.

Les API suivantes sont utiles pour suivre des modèles dans le cadre de votre flux de travail de suivi des expériences. Utilisez les API présentées sur cette page pour journaliser des modèles dans un run et pour accéder aux métriques, aux tableaux, aux médias et à d’autres objets.W&B vous recommande d’utiliser W&B Artifacts si vous souhaitez :

Créer et suivre différentes versions de données sérialisées autres que des modèles, comme des jeux de données, des prompts, etc.
Explorer les graphes de traçabilité d’un modèle ou de tout autre objet suivi dans W&B.
Interagir avec les artifacts de modèle créés par ces méthodes, par exemple en mettant à jour leurs propriétés (métadonnées, alias et descriptions)

Pour plus d’informations sur W&B Artifacts et sur les cas d’usage avancés de gestion des versions, consultez la documentation Artifacts.

Journaliser un modèle dans un run

Utilisez log_model pour journaliser un artifact de modèle contenant les fichiers d’un répertoire que vous spécifiez. La méthode log_model marque également l’artifact de modèle obtenu comme sortie du run W&B. Si vous marquez un modèle comme entrée ou sortie d’un run W&B, vous pouvez suivre ses dépendances et ses associations. Consultez la traçabilité du modèle dans l’interface de l’application W&B. Pour plus d’informations, voir la page Explorer et parcourir les graphes d’artifacts du chapitre Artifacts. Transmettez au paramètre path le chemin où vos fichiers de modèle sont enregistrés. Ce chemin peut désigner un fichier local, un répertoire ou un URI de référence vers un bucket externe, par exemple s3://bucket/path. Veillez à remplacer les valeurs entre <> par les vôtres.

import wandb

# Initialiser un run W&B
with wandb.init(project="<your-project>", entity="<your-entity>") as run:

    # Journaliser le modèle
    run.log_model(path="<path-to-model>", name="<name>")

Vous pouvez, si vous le souhaitez, fournir un nom pour l’artifact de modèle via le paramètre name. Si name n’est pas spécifié, W&B utilise comme nom le nom de base (basename) du chemin d’entrée, précédé de l’ID du run.

Notez le name que vous attribuez au modèle, ou que W&B lui attribue. Vous en aurez besoin pour récupérer le chemin du modèle avec la méthode wandb.Run.use_model().

Consultez log_model dans la Référence de l’API pour connaître les paramètres.

Télécharger et utiliser un modèle journalisé

Utilisez la fonction use_model pour accéder aux fichiers de modèle précédemment journalisés dans un run W&B et les télécharger. Indiquez le nom de l’artifact de modèle dans lequel sont stockés les fichiers de modèle à récupérer. Ce nom doit correspondre à celui d’un artifact de modèle journalisé existant. Si vous n’avez pas défini name lors de la journalisation initiale des fichiers avec log_model, le nom attribué par défaut est le nom de base du chemin d’entrée, précédé de l’ID du run. Veillez à remplacer les valeurs entre <> par les vôtres :

import wandb

# Initialiser un run
with wandb.init(project="<your-project>", entity="<your-entity>") as run:

    # Accéder au modèle et le télécharger. Renvoie le chemin de l’artifact téléchargé
    downloaded_model_path = run.use_model(name="<your-model-name>")

La fonction use_model renvoie le chemin des fichiers du modèle téléchargé. Conservez ce chemin si vous souhaitez lier ce modèle ultérieurement. Dans l’extrait de code précédent, le chemin renvoyé est stocké dans une variable nommée downloaded_model_path. Pour connaître les paramètres et le type de retour, consultez use_model dans la référence de l’API.

​Journaliser des modèles

​Journaliser un modèle dans un run

​Télécharger et utiliser un modèle journalisé

Journaliser des modèles

Journaliser un modèle dans un run

Télécharger et utiliser un modèle journalisé