Fix max number of frames per episode

Kaixhin · Oct 20, 2018 · de04b85 · de04b85
1 parent da43d9a
commit de04b85
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)