このページでは、W&B を OpenAI Gym とともに使用して、強化学習環境の動画を自動的に取得する方法について説明します。W&B でエージェントの動作と実験のメトリクスを並べて確認できます。
OpenAI Gym を使用する場合、W&B は
gym.wrappers.Monitor によって生成された環境の動画を自動的にログします。これを有効にするには、wandb.init() の monitor_gym キーワード引数を True に設定するか、wandb.gym.monitor() を呼び出します。
gym 統合は軽量です。gym からログされる動画ファイルの名前を確認し、それにちなんで名前を付け、一致するものがない場合は "videos" にフォールバックします。より細かい制御が必要な場合は、手動で動画をログできます。
CleanRL による OpenRL Benchmark は、OpenAI Gym のサンプルでこの統合を使用しています。gym を W&B とともに使用する方法を示すソースコード (特定の run で使用されたコードを含む) をご覧いただけます。
