From de04b85dc136ba60fe5b9cede3a7a772e021ccac Mon Sep 17 00:00:00 2001 From: Kaixhin Date: Sat, 20 Oct 2018 16:26:51 +0100 Subject: [PATCH] Fix max number of frames per episode --- env.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/env.py b/env.py index f53e7a1..80d1753 100644 --- a/env.py +++ b/env.py @@ -10,7 +10,7 @@ def __init__(self, args): self.device = args.device self.ale = atari_py.ALEInterface() self.ale.setInt('random_seed', args.seed) - self.ale.setInt('max_num_frames', args.max_episode_length) + self.ale.setInt('max_num_frames_per_episode', args.max_episode_length) self.ale.setFloat('repeat_action_probability', 0) # Disable sticky actions self.ale.setInt('frame_skip', 0) self.ale.setBool('color_averaging', False)