Reference an artifact version with aliases

​

デフォルトのエイリアス

• コレクションにリンクした最新のアーティファクトのバージョンには、latest エイリアスを割り当てます。
• 一意のバージョン番号を割り当てます。W&B は、リンクしたアーティファクトのバージョンをゼロから数えます。W&B は、この数値を使用して、そのアーティファクトに一意のバージョン番号を割り当てます。

​

カスタムエイリアス

• dataset_version_v0、dataset_version_v1、dataset_version_v2 などのエイリアスを使用して、モデルがどのデータセットでトレーニングされたかを特定できます。
• best_model エイリアスを使用して、最もパフォーマンスの高いアーティファクトのモデルバージョンを追跡できます。

import wandb

# run を初期化します
with wandb.init(entity = "<team_entity>", project = "<project_name>") as run:

    # アーティファクト オブジェクトを作成します
    # type パラメーターはアーティファクト オブジェクトのタイプと
    # コレクションのタイプの両方を指定します
    artifact = wandb.Artifact(name = "<name>", type = "<type>")

    # アーティファクト オブジェクトにファイルを追加します。
    # ローカルマシン上のファイルへのパスを指定します。
    artifact.add_file(local_path = "<local_path_to_artifact>")

    # アーティファクトをリンクするコレクションと registry を指定します
    REGISTRY_NAME = "<registry_name>"
    COLLECTION_NAME = "<collection_name>"
    target_path=f"wandb-registry-{REGISTRY_NAME}/{COLLECTION_NAME}"

    # artifact version をコレクションにリンクします
    # この artifact version に 1 つ以上のエイリアスを追加します
    run.link_artifact(
        artifact = artifact, 
        target_path = target_path, 
        aliases = ["<alias_1>", "<alias_2>"]
        )

​

保護されたエイリアス

• 本番: アーティファクトのバージョンが本番で使用できる準備完了の状態です。
• ステージング: アーティファクトのバージョンがテストできる準備完了の状態です。

​

保護されたエイリアスを作成する

1. https://forge.coreweave.com/registry/ の Registry にアクセスします。
2. Registry を選択します。
3. ページの右上にある歯車ボタンをクリックして、Registry の設定を表示します。
4. Protected Aliases セクションで、+ ボタンをクリックして、保護されたエイリアスを 1 つ以上追加します。

1. https://forge.coreweave.com/registry/ の Registry にアクセスします。
2. コレクションの View details ボタンをクリックします。
3. Versions セクションで、特定のアーティファクトのバージョンの View ボタンを選択します。
4. Aliases フィールドの横にある + ボタンをクリックして、保護されたエイリアスを 1 つ以上追加します。

​

既存のエイリアスを検索する

1. https://forge.coreweave.com/registry/ の Registry にアクセスします。
2. ページ上部の検索バーに検索語を入力します。Enter キーを押して検索します。

​

例

import wandb

# 実行を初期化します
with wandb.init(entity = "smle-reg-team-2", project = "zoo_experiment") as run:

    # アーティファクトオブジェクトを作成します
    # typeパラメーターは、アーティファクトオブジェクトのタイプと
    # コレクションのタイプの両方を指定します
    artifact = wandb.Artifact(name = "zoo_dataset", type = "dataset")

    # アーティファクトオブジェクトにファイルを追加します。
    # ローカルマシン上のファイルへのパスを指定します。
    artifact.add_file(local_path="zoo_dataset.pt", name="zoo_dataset")
    artifact.add_file(local_path="zoo_labels.pt", name="zoo_labels")

    # アーティファクトをリンクするコレクションと Registry を指定します
    REGISTRY_NAME = "Model"
    COLLECTION_NAME = "Zoo_Classifier_Models"
    target_path=f"wandb-registry-{REGISTRY_NAME}/{COLLECTION_NAME}"

    # アーティファクトバージョンをコレクションにリンクします
    # このアーティファクトバージョンに1つ以上のエイリアスを追加します
    run.link_artifact(
        artifact = artifact,
        target_path = target_path,
        aliases = ["production-us", "production-eu"]
        )

1. まず、アーティファクト オブジェクト (wandb.Artifact()) を作成します。
2. 次に、wandb.Artifact.add_file() を使用して、2 つのデータセット PyTorch テンソルをアーティファクト オブジェクトに追加します。
3. 最後に、link_artifact() を使用して、アーティファクトのバージョンを Zoo_Classifier_Models Registry 内の Model コレクションにリンクします。また、aliases パラメーターに production-us と production-eu を引数として渡し、アーティファクトのバージョンに 2 つのカスタム エイリアスを追加します。