wandb.init() をコンテキストマネージャーとして使用することです。これにより、Run が自動的に終了し、スクリプト で例外が発生した場合には失敗としてマークされます。
run.finish() を呼び出すことも可能です。
複数のアクティブな Run
wandb 0.19.10 以降では、reinit 設定を "create_new" に設定することで、複数の Run を同時にアクティブにすることが可能です。
reinit="create_new" に関する詳細や、W&B インテグレーション に関する注意事項については、Multiple runs per process を参照してください。