训练gym游戏时记录视频

最新推荐文章于 2024-12-22 02:07:15 发布

ok_kakaka

最新推荐文章于 2024-12-22 02:07:15 发布

阅读量1.8k

点赞数 2

本文链接：https://blog.csdn.net/clksjx/article/details/103092329

版权

from gym import wrappers, logger


env_id = 'CartPole-v0'

logger.setLevel(logger.ERROR)

env = gym.make(env_id)

outdir = "/tmp/dqn-%s" % env_id

env = wrappers.Monitor(env, directory=outdir, video_callable=lambda episode_id: episode_id % 100 == 0, force=True)  
# video_callable可以设置多少episode记录一次 
#（video_callable=lambda episode_id: True）[reference](https://github.com/openai/gym/issues/494)
# force可强制性地覆盖掉之前生成的记录文件


# train....

env.close()  # close env and write monitor result info to disk