モデルバージョンをダウンロードする

W&B Python SDK で モデル をダウンロードする方法

W&B Python SDK を使用して、Model Registry にリンクしたモデルアーティファクトをダウンロードします。

<> の中の値を自身のものに置き換えてください:

import wandb

# Run を初期化
run = wandb.init(project="<project>", entity="<entity>")

# モデルへのアクセスとダウンロード。ダウンロードしたアーティファクトへのパスを返します
downloaded_model_path = run.use_model(name="<your-model-name>")
Python

モデルバージョンを以下のいずれかの形式で参照します:

  • latest - 最も最近リンクされたモデルバージョンを指定するために latest エイリアスを使用します。
  • v# - 特定のバージョンを取得するために v0v1v2 などを使用します。
  • alias - モデルバージョンに対してチームが設定したカスタムエイリアスを指定します。

API リファレンスガイドの use_model を参照して、使用可能なパラメータと返り値の型についての詳細を確認してください。

例:ログされたモデルをダウンロードして使用する

例えば、以下のコードスニペットでは、ユーザーが use_model API を呼び出しています。彼らは取得したいモデルアーティファクトの名前を指定し、さらにバージョン/エイリアスも提供しています。その後、API から返されたパスを downloaded_model_path 変数に格納しています。

import wandb

entity = "luka"
project = "NLP_Experiments"
alias = "latest"  # モデルバージョンのセマンティックニックネームまたは識別子
model_artifact_name = "fine-tuned-model"

# Run を初期化
run = wandb.init()
# モデルへのアクセスとダウンロード。ダウンロードしたアーティファクトへのパスを返します

downloaded_model_path = run.use_model(name=f"{entity/project/model_artifact_name}:{alias}")
Python

<> の中の値を自身のものに置き換えてください:

import wandb
# Run を初期化
run = wandb.init(project="<project>", entity="<entity>")
# モデルへのアクセスとダウンロード。ダウンロードしたアーティファクトへのパスを返します
downloaded_model_path = run.use_model(name="<your-model-name>")
Python

モデルバージョンを以下のいずれかの形式で参照します:

  • latest - 最も最近リンクされたモデルバージョンを指定するために latest エイリアスを使用します。
  • v# - 特定のバージョンを取得するために v0v1v2 などを使用します。
  • alias - モデルバージョンに対してチームが設定したカスタムエイリアスを指定します。

API リファレンスガイドの use_model を参照して、使用可能なパラメータと返り値の型についての詳細を確認してください。

  1. https://wandb.ai/registry/model の Model Registry App に移動します。
  2. ダウンロードしたいモデルを含む登録済みモデル名の隣にある 詳細を見る を選択します。
  3. バージョンセクション内で、ダウンロードしたいモデルバージョンの隣にある表示ボタンを選択します。
  4. ファイル タブを選択します。
  5. ダウンロードしたいモデルファイルの隣にあるダウンロードボタンをクリックします。