def create_env(conf):
env = gym.make(conf['env'])
if conf['monitor_dir']:
env = wrappers.Monitor(env, conf['monitor_dir'], force=True)
if conf['use_atari_wrapper']:
env = AtariRescale42x42Wrapper(env)
env = NormalizeWrapper(env)
return env
评论列表
文章目录