- W&B にログインする
W&B ライブラリをインポートし、W&B にログインします。まだサインアップしていない場合は、無料の W&B アカウントにサインアップする必要があります。
- run を初期化する
wandb.init() を使用して run を初期化します。これにより、データを同期し、ログするためのバックグラウンドプロセスが生成されます。プロジェクト名とジョブタイプを指定します:
- アーティファクト オブジェクトを作成する
wandb.Artifact() を使用してアーティファクト オブジェクトを作成します。name パラメーターと type パラメーターに、それぞれアーティファクトの名前とファイルタイプの説明を指定してください。
例えば、次のスニペットは ‘bicycle-dataset’ というアーティファクトを ‘dataset’ というラベルで作成する方法を示しています:
- データセットをアーティファクトに追加する
アーティファクトにファイルを追加します。一般的なファイルタイプにはモデルとデータセットが含まれます。次の例では、ローカルに保存されている dataset.h5 という名前のデータセットをアーティファクトに追加します:
dataset.h5 というファイル名を、アーティファクトに追加したいファイルのパスに置き換えてください。
- データセットをログする
W&B の Run オブジェクトの wandb.Run.log_artifact() メソッドを使用して、アーティファクトのバージョンを保存し、run の出力 としてアーティファクトを宣言します。
'latest' エイリアス は、アーティファクトをログするときにデフォルトで作成されます。アーティファクトのエイリアスとバージョンに関する詳細は、それぞれ カスタムエイリアスを作成する と 新しいアーティファクトのバージョンを作成する を参照してください。
これらをまとめると、これまでのスクリプトは次のようになります:
- アーティファクトをダウンロードして使用する
以下のコード例は、W&B サーバーにログして保存したアーティファクトを使用するために実行できる手順を示しています。
- まず、
wandb.init()を使用して新しい run オブジェクトを初期化します。 - 次に、run オブジェクトの
wandb.Run.use_artifact()メソッドを使用して、使用するアーティファクトを W&B に指示します。これによりアーティファクト オブジェクトが返されます。 - 最後に、アーティファクトの
wandb.Artifact.download()メソッドを使用して、アーティファクトの内容をダウンロードします。
wandb.Api) を使用して、run の外部で W&B にすでに保存されているデータをエクスポート (または更新) することもできます。詳細については、外部ファイルをトラッキングする を参照してください。