Créer un alias d'artifact

Créez et gérez des alias personnalisés pour faire référence à des versions spécifiques d’artifact W&B avec des noms explicites comme best ou production.

Utilisez les alias comme pointeurs vers des versions spécifiques. Par défaut, wandb.Run.log_artifact() ajoute l’alias latest à la version enregistrée.

W&B crée une version d’artifact v0 et l’associe à votre artifact lorsque vous enregistrez cet artifact pour la première fois. W&B calcule une somme de contrôle du contenu lorsque vous l’enregistrez à nouveau dans le même artifact. Si l’artifact a changé, W&B enregistre une nouvelle version v1.

Par exemple, si vous voulez que votre script d’entraînement récupère la version la plus récente d’un jeu de données, indiquez latest lorsque vous utilisez cet artifact. L’exemple de code suivant montre comment télécharger un artifact de jeu de données récent nommé bike-dataset avec l’alias latest :

import wandb

with wandb.init(project="<project>") as run:
    artifact = run.use_artifact("bike-dataset:latest")
    artifact.download()

Vous pouvez également attribuer un alias personnalisé à une version d’artifact. Par exemple, si vous voulez indiquer que ce point de contrôle du modèle est le meilleur selon la métrique AP-50, vous pouvez ajouter la chaîne 'best-ap50' comme alias lorsque vous journalisez l’artifact du modèle.

with wandb.init(project="<project>") as run:
    artifact = wandb.Artifact("run-3nq3ctyy-bike-model", type="model")
    artifact.add_file("model.h5")
    run.log_artifact(artifact, aliases=["latest", "best-ap50"])