Cette page explique comment utiliser W&B avec OpenAI Gym pour capturer automatiquement des vidéos de vos environnements d’apprentissage par renforcement. Vous pouvez ensuite examiner le comportement de l’agent en parallèle des métriques de votre expérience dans W&B.
Si vous utilisez OpenAI Gym, W&B enregistre automatiquement les vidéos de votre environnement générées par
gym.wrappers.Monitor. Pour l’activer, définissez l’argument mot-clé monitor_gym de wandb.init() sur True, ou appelez wandb.gym.monitor().
L’intégration gym est légère. Elle regarde le nom du fichier vidéo enregistré depuis gym et lui donne ce nom, ou utilise "videos" par défaut si elle ne trouve pas de correspondance. Si vous voulez plus de contrôle, vous pouvez enregistrer une vidéo manuellement.
Le OpenRL Benchmark de CleanRL utilise cette intégration pour ses exemples OpenAI Gym. Vous pouvez trouver le code source (y compris le code spécifique utilisé pour certains runs) qui montre comment utiliser gym avec W&B.
