メインコンテンツへスキップ
wandb から利用できる W&B Python SDK を使用すると、モデルのトレーニングやファインチューニングを行い、実験から本番までモデルを管理できます。
この SDK でトレーニングとファインチューニングを行った後は、Public API を使用してログされたデータをクエリして分析し、Reports and Workspaces API を使用して、作業内容を要約した Web で公開可能なレポートを生成できます。

インストールと設定

サインアップしてAPIキーを発行する

W&Bでこのマシンを認証するには、まず User Settings でAPIキーを発行する必要があります。

パッケージのインストールとインポート

W&B ライブラリをインストールします。
pip install wandb

W&B Python SDK をインポートする:

import wandb

# チーム entityを指定する
entity = "<team_entity>"

# runが記録されるproject
project = "my-awesome-project"

with wandb.init(entity=entity, project=project) as run:
   run.log({"accuracy": 0.9, "loss": 0.1})