メインコンテンツまでスキップ

モデルバージョンをリンクする

モデルバージョンをW&Bの登録済みモデルにリンクするには、W&BアプリやPython SDKを使ってプログラム的に行うことができます。

プログラム的にモデルをリンクする

link_model メソッドを使って、W&Bのrunにモデルファイルをプログラム的にログし、W&B Model Registryにリンクします。

<> で囲まれた他の値を自分のものに置き換えてください:

import wandb

run = wandb.init(entity="<entity>", project="<project>")
run.link_model(path="<path-to-model>", registered_model_name="<registered-model-name>")
run.finish()

registered-model-name パラメータに指定した名前が既に存在しない場合、W&Bはあなたのために登録済みモデルを作成します。

例えば、"Fine-Tuned-Review-Autocompletion"(registered-model-name="Fine-Tuned-Review-Autocompletion")という名前の登録済みモデルがあり、いくつかのモデルバージョン (v0, v1, v2) がリンクされているとします。新しいモデルをプログラム的にリンクし、同じ登録済みモデル名(registered-model-name="Fine-Tuned-Review-Autocompletion")を使用すると、W&Bはこのモデルを既存の登録済みモデルにリンクし、モデルバージョン v3 を割り当てます。この名前の登録済みモデルが存在しない場合、新しい登録済みモデルが作成され、モデルバージョン v0 を持ちます。

"Fine-Tuned-Review-Autocompletion" 登録済みモデルの例はこちら

インタラクティブにモデルをリンクする

Model Registry もしくは Artifact ブラウザを使ってインタラクティブにモデルをリンクします。

  1. https://wandb.ai/registry/model にある Model Registry App に移動します。
  2. 登録済みモデルの名前の横にマウスをホバーします。
  3. View details アイコン(3点リーダー)を選択します。
  4. ドロップダウンから Link new version を選択します。
  5. Project ドロップダウンから、モデルを含むプロジェクトの名前を選択します。
  6. Model Artifact ドロップダウンから、モデルアーティファクトの名前を選択します。
  7. Version ドロップダウンから、登録済みモデルにリンクしたいモデルバージョンを選択します。

リンクされたモデルのソースを見る

リンクされたモデルのソースを見る方法は2つあります: モデルがログされたプロジェクト内のアーティファクトブラウザとW&B Model Registryです。

ポインタはモデルレジストリ内の特定のモデルバージョンを、モデルがログされたプロジェクト内にあるソースモデルアーティファクトに接続します。ソースモデルアーティファクトにもモデルレジストリへのポインタがあります。

  1. あなたのモデルレジストリに移動します: https://wandb.ai/registry/model
  2. 登録済みモデルの名前の横にある View details を選択します。
  3. Versions セクション内で、調査したいモデルバージョンの横にある View を選択します。
  4. 右パネル内の Version タブをクリックします。
  5. Version overview セクション内に Source Version フィールドを持つ行があります。これはモデルの名前とモデルのバージョンの両方を表示します。

例えば、下記画像では v0 モデルバージョンで呼ばれる mnist_model ( Source version フィールド mnist_model:v0 を参照) が、 MNIST-dev という登録モデルにリンクされています。

Was this page helpful?👍👎