pip install を 1 回実行するだけで利用できます。
W&B のすべてのログ機能は、PyTorch DDP などのデータ並列マルチ GPU トレーニングで利用可能です。
コア実験をトラッキングする
まず、YOLOv5 と一緒に W&B Python SDK (wandb) をインストールし、通常どおりトレーニングを実行してください。wandb をインストールすると、W&B に組み込まれたログ機能が有効化され、システムメトリクス、モデルのメトリクス、メディアがインタラクティブなダッシュボードにログされます。
インテグレーションをカスタマイズする
実験管理ができるようになったら、YOLO にいくつかのコマンドライン引数を渡すことで、モデルやデータセットのバージョン管理、予測の可視化といった W&B の追加機能を有効にできます。--save_periodに数値を指定すると、W&B はsave_periodエポックごとの終了時に モデルのバージョン を保存します。このバージョンにはモデルの重みが含まれ、検証セットで最も性能の高いモデルにタグが付けられます。--upload_datasetフラグを有効にすると、バージョン管理のためにデータセットもアップロードされます。--bbox_intervalに数値を指定すると、データの可視化が有効になります。W&B はbbox_intervalエポックごとの終了時に、検証セットに対するモデルの出力をアップロードします。
- モデルのバージョン管理のみ
- モデルのバージョン管理とデータの可視化
すべての W&B アカウントで、データセットとモデル用に 100 GB の無料ストレージを利用できます。


データとモデルのバージョン管理を使用すると、一時停止した実験やクラッシュした実験を、セットアップせずにどのデバイスからでも再開できます。詳細は Colab を参照してください。