Registry

W&B Registry は、組織内の artifact バージョンの厳選された中央リポジトリです。組織内で許可を持っている ユーザー は、所属する Teams に関係なく、すべての artifact のライフサイクルをダウンロード、共有、および共同で管理できます。

Registry を使用して、artifact の バージョン を追跡し、artifact の使用状況と変更の履歴を監査し、artifact のガバナンスとコンプライアンスを確保し、モデル CI/CD などのダウンストリーム プロセス を自動化できます。

要約すると、W&B Registry を使用して以下を行います。

• 機械学習タスクを満たす artifact バージョン を組織内の他の ユーザー にプロモートします。
• 特定の artifact を見つけたり参照したりできるように、タグを使用して artifact を整理します。
• artifact の リネージ を追跡し、変更の履歴を監査します。
• モデル CI/CD などのダウンストリーム プロセス を自動化します。
• 組織内で誰が各 registry 内の artifact に アクセス できるかを制限します。

上の図は、“Model” および “Dataset” コア registry とカスタム registry を含む Registry App を示しています。

基本を学ぶ

各組織には、モデル および データセット artifact を整理するために使用できる Models および Datasets という 2 つの registry が最初から含まれています。組織のニーズに基づいて他の artifact タイプ を整理するために、追加の registry を作成できます。

各registryは、1 つ以上のコレクションで構成されています。各コレクションは、明確なタスクまたは ユースケース を表します。

artifact を registry に追加するには、まず特定の artifact バージョン を W&B に ログ します。artifact を ログ するたびに、W&B はその artifact に バージョン を自動的に割り当てます。artifact バージョン は 0 からインデックスを開始するため、最初の バージョン は v0、2 番目の バージョン は v1 のようになります。

artifact を W&B に ログ したら、その特定の artifact バージョン を registry 内のコレクションにリンクできます。

例として、次の コード 例は、“my_model.txt” という名前のモデル artifact をコア registryの “first-collection” という名前のコレクションに ログ してリンクする方法を示しています。

1. W&B run を初期化します。
2. artifact を W&B に ログ します。
3. artifact バージョン をリンクするコレクションと registry の名前を指定します。
4. artifact をコレクションにリンクします。

この Python コード を スクリプト に保存して実行します。W&B Python SDK バージョン 0.18.6 以降が必要です。

import wandb
import random

# Artifact を追跡するために W&B run を初期化します。
run = wandb.init(project="registry_quickstart") 

# ログ できるようにシミュレートされたモデルファイルを作成します
with open("my_model.txt", "w") as f:
   f.write("Model: " + str(random.random()))

# Artifact を W&B に ログ します
logged_artifact = run.log_artifact(
    artifact_or_path="./my_model.txt", 
    name="gemma-finetuned", 
    type="model" # artifact タイプを指定します
)

# artifact を公開するコレクションと registry の名前を指定します
COLLECTION_NAME = "first-collection"
REGISTRY_NAME = "model"

# artifact を registry にリンクします
run.link_artifact(
    artifact=logged_artifact, 
    target_path=f"wandb-registry-{REGISTRY_NAME}/{COLLECTION_NAME}"
)

返された run オブジェクトの link_artifact(target_path = "") メソッドで指定したコレクションが、指定した registry 内に存在しない場合、W&B は自動的にコレクションを作成します。

Registry App に移動して、 ユーザー と組織の他のメンバーが公開する artifact バージョン を表示します。これを行うには、まず W&B に移動します。[アプリケーション] の下の左側のサイドバーで Registry を選択します。“Model” registry を選択します。registry 内に、リンクされた artifact バージョン を持つ “first-collection” コレクションが表示されます。

artifact バージョン を registry 内のコレクションにリンクすると、組織のメンバーは、適切な権限を持っている場合、artifact バージョン の表示、ダウンロード、管理、ダウンストリーム オートメーション の作成などを行うことができます。

W&B Registry を有効にする

デプロイメントタイプに基づいて、次の条件を満たして W&B Registry を有効にします。

開始するためのリソース

ユースケース に応じて、W&B Registry を開始するための次のリソースをご覧ください。

• チュートリアルビデオをご覧ください。
  • Weights & Biases から Registry を開始する
• W&B の Model CI/CD コースを受講して、次の方法を学びましょう。
  • W&B Registry を使用して、artifact の管理と バージョン 管理、リネージ の追跡、さまざまなライフサイクルステージでのモデル のプロモーションを行います。
  • Webhook を使用して、モデル 管理 ワークフロー を自動化します。
  • registry を外部 ML システム および ツール と統合して、モデル の 評価 、監視、および デプロイメント を行います。

レガシー Model Registry から W&B Registry への移行

レガシー Model Registry は、正確な日付はまだ決定されていませんが、廃止が予定されています。レガシー Model Registry を廃止する前に、W&B はレガシー Model Registry のコンテンツを W&B Registry に移行します。

レガシー Model Registry から W&B Registry への移行 プロセス の詳細については、レガシー Model Registry からの移行を参照してください。

移行が行われるまで、W&B はレガシー Model Registry と新しい Registry の両方をサポートします。

レガシー Model Registry を表示するには、W&B App の Model Registry に移動します。ページの上部に、レガシー Model Registry App UI の使用を有効にするバナーが表示されます。

ご質問がある場合、または移行に関する懸念について W&B Product Team と話したい場合は、support@wandb.com までご連絡ください。