モデルをログする
以下のガイドでは、W&B run にモデルをログする方法と、それらを操作する方法について説明します。以下の API は、実験管理ワークフローの一部としてモデルを追跡するのに役立ちます。このページに記載されている API を使用して、run にモデルをログし、メトリクス、表、メディア、その他のオブジェクトにアクセスします。W&B では、以下のような場合に W&B Artifacts の使用をおすすめします:
- データセットやプロンプトなど、モデル以外のシリアライズされたデータのさまざまなバージョンを作成および追跡する場合。
- W&B で追跡されたモデルまたはその他のオブジェクトの リネージ グラフ を探索する場合。
- これらのメソッドが作成したモデル アーティファクトを操作する場合、たとえば プロパティの更新 (メタデータ、エイリアス、説明) など。
モデルを run にログする
log_model を使用して、指定したディレクトリ内のコンテンツを含むモデル アーティファクトをログします。log_model メソッドは、作成されたモデル アーティファクトを W&B run の出力としてもマークします。
モデルを W&B run の入力または出力としてマークすると、モデルの依存関係や関連付けをトラッキングできます。W&B のアプリ UI でモデルのリネージを確認できます。詳細については、Artifacts の章にあるアーティファクト グラフの探索と走査ページを参照してください。
モデルファイルの保存先のパスを path パラメーターに指定します。パスには、ローカルファイル、ディレクトリ、または s3://bucket/path のような外部バケットへの参照 URI を指定できます。
<> で囲まれた値は、必ずご自身の値に置き換えてください。
name パラメーターにモデル アーティファクトの名を指定します。name を指定しない場合、W&B は入力パスの basename の先頭に run ID を付けたものを名として使用します。
自分または W&B がモデルに割り当てた
name を控えておいてください。wandb.Run.use_model() メソッドでモデルのパスを取得する際に、モデル名が必要になります。log_model を参照してください。
ログされたモデルのダウンロードと使用
use_model 関数を使用して、以前に W&B run にログされたモデルファイルにアクセスしてダウンロードします。
取得したいモデルファイルが保存されているモデル アーティファクトの名を指定します。指定する名は、既存のログされたモデル アーティファクトの名と一致する必要があります。
log_model でファイルを元にログする際に name を定義しなかった場合、割り当てられるデフォルトの名は、入力パスの basename に run ID を前置したものです。
<> で囲まれた値を自分の値に置き換えてください：
downloaded_model_path という変数に格納されています。
パラメーターと戻り値のタイプについては、API リファレンスの use_model を参照してください。