From b0a521a3863d08c6b0b1a26d8f494e066f4403fb Mon Sep 17 00:00:00 2001 From: rajfly Date: Fri, 2 Aug 2024 03:48:50 +0800 Subject: [PATCH] default max_episode_steps to 27K (108K frames) --- baselines/common/atari_wrappers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/baselines/common/atari_wrappers.py b/baselines/common/atari_wrappers.py index 8be0a2f254..2039375ad4 100644 --- a/baselines/common/atari_wrappers.py +++ b/baselines/common/atari_wrappers.py @@ -263,7 +263,7 @@ def count(self): def frame(self, i): return self._force()[..., i] -def make_atari(env_id, max_episode_steps=None): +def make_atari(env_id, max_episode_steps=27000): env = gym.make(env_id) assert 'NoFrameskip' in env.spec.id env = NoopResetEnv(env, noop_max=30)