Hotfix param

hill-a · Oct 2, 2018 · 8c43ec3 · 8c43ec3
1 parent 582a852
commit 8c43ec3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/stable_baselines/acer/run_atari.py b/stable_baselines/acer/run_atari.py
@@ -19,7 +19,7 @@ def train(env_id, num_timesteps, seed, policy, lr_schedule, num_cpu):
                                  'double_linear_con', 'middle_drop' or 'double_middle_drop')
     :param num_cpu: (int) The number of cpu to train on
     """
-    env = VecFrameStack(make_atari_env(env_id, num_env, seed), 4)
+    env = VecFrameStack(make_atari_env(env_id, num_cpu, seed), 4)
     if policy == 'cnn':
         policy_fn = CnnPolicy
     elif policy == 'lstm':