OpenAI Gym

W&B를 OpenAI Gym과 인테그레이션하는 방법.

“2021년부터 Gym을 유지 관리해 온 팀은 모든 향후 개발을 Gymnasium으로 이전했습니다. Gymnasium은 Gym과 100% 호환되는 대체 라이브러리(import gymnasium as gym)이며, Gym은 앞으로 더 이상 업데이트를 받지 않습니다.” (출처)

Gym은 더 이상 활발히 유지 관리되는 프로젝트가 아니므로, Gymnasium과의 인테그레이션을 사용해 보세요.

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에 사용된 코드 포함)를 확인할 수 있습니다.