Update cleanrl/ppo_atari_accelerate.py

Co-authored-by: Costa Huang <costa.huang@outlook.com>
vwxyzjn · Feb 13, 2024 · ea6fc99 · ea6fc99
1 parent f7cd970
commit ea6fc99
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/cleanrl/ppo_atari_accelerate.py b/cleanrl/ppo_atari_accelerate.py
@@ -92,6 +92,8 @@ class Args:
     """the mini-batch size (computed in runtime)"""
     num_iterations: int = 0
     """the number of iterations (computed in runtime)"""
+    world_size: int = 0
+    """the number of processes (computed in runtime)"""
 
 
 def make_env(env_id, idx, capture_video, run_name):