artifact エイリアスを作成する

best や production のような意味のある名前で、特定の W&B artifact バージョンを参照するためのカスタムエイリアスを作成して管理します。

エイリアスは、特定のバージョンを指すポインタとして使用します。デフォルトでは、wandb.Run.log_artifact() はログしたバージョンに latest エイリアスを追加します。

W&B は、artifact を初めてログしたときに artifact バージョン v0 を作成し、その artifact に関連付けます。同じ artifact に再度ログすると、W&B は内容のチェックサムを確認します。artifact に変更があれば、W&B は新しいバージョン v1 を保存します。

たとえば、トレーニングスクリプトでデータセットの最新バージョンを取得したい場合は、その artifact を使用するときに latest を指定します。次のコード例は、latest エイリアスが付いた bike-dataset という名前の最新のデータセット artifact をダウンロードする方法を示しています。

import wandb

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

artifact バージョンにカスタム エイリアスを付けることもできます。たとえば、そのモデル チェックポイントがメトリクス AP-50 で最良であることを示したい場合は、モデル artifact をログするときに、文字列 'best-ap50' をエイリアスとして追加できます。

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"])