示例#1
0
 def _thunk():
   env = make_atari(env_id)
   env.seed(seed + rank)
   if use_monitor:
     env = Monitor(env, logger.get_dir() and os.path.join(logger.get_dir(),
                                                          str(rank)))
   return wrap_deepmind(env, **wrapper_kwargs)
示例#2
0
def create_single_atari_env(env_name, seed, use_monitor, split=''):
  env = atari_wrappers.make_atari(env_name)
  env.seed(seed)
  if use_monitor:
    env = Monitor(
        env, logger.get_dir() and os.path.join(logger.get_dir(), str(seed)))
  env = atari_wrappers.wrap_deepmind(env, frame_stack=True)
  return env