From 3bbf3f351150df7ab595a1de0487319a69146edf Mon Sep 17 00:00:00 2001 From: Peter Zhokhov Date: Thu, 30 Aug 2018 09:40:42 -0700 Subject: [PATCH] allow_early_resets=True in create_vec_env --- baselines/common/cmd_util.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/baselines/common/cmd_util.py b/baselines/common/cmd_util.py index f20809e..db395a7 100644 --- a/baselines/common/cmd_util.py +++ b/baselines/common/cmd_util.py @@ -29,7 +29,10 @@ def make_vec_env(env_id, env_type, num_env, seed, wrapper_kwargs=None, start_ind def _thunk(): env = make_atari(env_id) if env_type == 'atari' else gym.make(env_id) env.seed(seed + 10000*mpi_rank + rank if seed is not None else None) - env = Monitor(env, logger.get_dir() and os.path.join(logger.get_dir(), str(mpi_rank) + '.' + str(rank))) + env = Monitor(env, + logger.get_dir() and os.path.join(logger.get_dir(), str(mpi_rank) + '.' + str(rank)) + allow_early_resets=True) + if env_type == 'atari': return wrap_deepmind(env, **wrapper_kwargs) elif reward_scale != 1: return RewardScaler(env, reward_scale) else: return env