モデルバージョンをリンクする
モデルバージョンを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 ブラウザを使ってインタラクティブにモデルをリンクします。
- Model Registry
- Artifact browser
- https://wandb.ai/registry/model にある Model Registry App に移動します。
- 登録済みモデルの名前の横にマウスをホバーします。
- View details アイコン(3点リーダー)を選択します。
- ドロップダウンから Link new version を選択します。
- Project ドロップダウンから、モデルを含むプロジェクトの名前を選択します。
- Model Artifact ドロップダウンから、モデルアーティファクトの名前を選択します。
- Version ドロップダウンから、登録済みモデルにリンクしたいモデルバージョンを選択します。
- あなたのプロジェクトのアーティファクト ブラウザに移動します:
https://wandb.ai/<entity>/<project>/artifacts
- 左側のサイドバーでアーティファクトアイコンを選択します。
- 登録にリンクしたいモデルバージョンをクリックします。
- Version overview セクション内で Link to registry ボタンをクリックします。
- 画面の右側に表示されるモーダルから、 Select a register model メニューのドロップダウンから、登録済みモデルを選択します。
- Next step をクリックします。
- (任意)Aliases ドロップダウンからエイリアスを選択します。
- Link to registry をクリックします。
リンクされたモデルのソースを見る
リンクされたモデルのソースを見る方法は2つあります: モデルがログされたプロジェクト内のアーティファクトブラウザとW&B Model Registryです。
ポインタはモデルレジストリ内の特定のモデルバージョンを、モデルがログされたプロジェクト内にあるソースモデルアーティファクトに接続します。ソースモデルアーティファクトにもモデルレジストリへのポインタがあります。
- Model Registry
- Artifact browser
- あなたのモデルレジストリに移動します: https://wandb.ai/registry/model
- 登録済みモデルの名前の横にある View details を選択します。
- Versions セクション内で、調査したいモデルバージョンの横にある View を選択します。
- 右パネル内の Version タブをクリックします。
- Version overview セクション内に Source Version フィールドを持つ行があります。これはモデルの名前とモデルのバージョンの両方を表示します。
例えば、下記画像では v0
モデルバージョンで呼ばれる mnist_model
( Source version フィールド mnist_model:v0
を参照) が、 MNIST-dev
という登録モデルにリンクされています。
- あなたのプロジェクトのアーティファクト ブラウザに移動します:
https://wandb.ai/<entity>/<project>/artifacts
- 左側のサイドバーでアーティファクトアイコンを選択します。
- アーティファクト パネルから model ドロップダウンメニューを展開します。
- モデルレジストリにリンクされた名前とバージョンのモデルを選択します。
- 右パネル内の Version タブをクリックします。
- Version overview セクション内に Linked To フィールドを持つ行があります。 これは登録済みモデルの名前とそのバージョン (
registered-model-name:version
) を表示します。
例えば、下記画像では MNIST-dev
という登録済みモデルがあります ( Linked To フィールドを参照)。バージョン v0
(mnist_model:v0
) のモデルバージョンが MNIST-dev
登録済みモデルを指しています。