OpenAI Gym

Intégrez W&B à OpenAI Gym pour suivre les expériences d’apprentissage par renforcement et enregistrer des vidéos des performances des épisodes.

“L’équipe qui maintient Gym depuis 2021 a transféré tout le développement futur vers Gymnasium, un remplaçant direct de Gym (import gymnasium as gym), et Gym ne recevra plus aucune mise à jour.” (Source)

Comme Gym n’est plus activement maintenu, essayez notre intégration avec Gymnasium.

Si vous utilisez OpenAI Gym, W&B enregistre automatiquement les vidéos de votre environnement générées par gym.wrappers.Monitor. Définissez simplement l’argument mot-clé monitor_gym de wandb.init sur True ou appelez wandb.gym.monitor().

Notre intégration Gym est très légère. Nous regardons simplement le nom du fichier vidéo enregistré depuis gym et lui donnons ce nom, ou utilisons "videos" par défaut si nous ne trouvons pas de correspondance. Si vous voulez plus de contrôle, vous pouvez toujours 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