実験の追跡( Experiments )やアーティファクト( Artifacts )の管理を行うために、W&B Python SDK の基本クラスを使用します。
機械学習の 実験 でこれらのクラスを使用する詳細については、 Experiments および Artifacts のドキュメントを参照してください。
Run
Artifact
Settings
import wandb # 新しい run を初期化します with wandb.init(project="my-experiments", config={"learning_rate": 0.001}) as run: # 設定にアクセスします config = run.config # トレーニング中のメトリクスをログに記録します for epoch in range(10): metrics = train_one_epoch() # トレーニングロジック run.log({ "loss": metrics["loss"], "accuracy": metrics["accuracy"], "epoch": epoch }) # サマリーメトリクスを記録します run.summary["best_accuracy"] = max_accuracy
import wandb with wandb.init(project="my-models") as run: # モデルをトレーニングします model = train_model() # モデルの Artifact を作成します model_artifact = wandb.Artifact( name="my-model", type="model", description="ResNet-50 trained on ImageNet subset", metadata={ "architecture": "ResNet-50", "dataset": "ImageNet-1K", "accuracy": 0.95 } ) # Artifact にモデルファイルを追加します model_artifact.add_file("model.pt") model_artifact.add_dir("model_configs/") # Artifact を W&B にログ記録します run.log_artifact(model_artifact)
import wandb # プログラムから設定を構成します wandb.Settings( project="production-runs", entity="my-team", mode="offline", # オフラインで実行し、後で同期します save_code=True, # ソースコードを保存します quiet=True # コンソール出力を抑制します ) # または環境変数を使用します # export WANDB_PROJECT=production-runs # export WANDB_MODE=offline # カスタム設定で初期化します with wandb.init() as run: # ここに実験のコードを記述します pass
import wandb with wandb.init(project="ml-pipeline") as run: # データセット Artifact を使用します dataset = run.use_artifact("dataset:v1") dataset_dir = dataset.download() # データセットを使用してモデルをトレーニングします model = train_on_dataset(dataset_dir) # データセットのリネージを持つモデル Artifact を作成します model_artifact = wandb.Artifact( name="trained-model", type="model" ) model_artifact.add_file("model.pt") # 自動的なリネージトラッキングを伴うログ記録を行います run.log_artifact(model_artifact)
このページは役に立ちましたか?