Cette page explique comment utiliser Weights & Biases 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 regard des métriques de votre expérience dans Weights & Biases.
Si vous utilisez OpenAI Gym, Weights & Biases consigne automatiquement les vidéos de votre environnement générées par
gym.wrappers.Monitor. Pour activer cette fonctionnalité, définissez l’argument nommé monitor_gym de wandb.init() sur True, ou appelez wandb.gym.monitor().
L’intégration gym est légère. Elle examine le nom du fichier vidéo journalisé depuis gym et nomme la vidéo d’après celui-ci, ou utilise "videos" par défaut si aucune correspondance n’est trouvée. Pour un contrôle plus fin, vous pouvez journaliser une vidéo manuellement.
L’OpenRL Benchmark de CleanRL utilise cette intégration pour ses exemples OpenAI Gym. Vous y trouverez du code source (y compris le code utilisé pour certains runs) montrant comment utiliser gym avec Weights & Biases.
