amend

pytorch · Apr 22, 2024 · 8d053b7 · 8d053b7
1 parent b3f7d70
commit 8d053b7
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tutorials/sphinx-tutorials/multiagent_competitive_ddpg.py b/tutorials/sphinx-tutorials/multiagent_competitive_ddpg.py
@@ -232,7 +232,7 @@
 n_evaders = 1
 n_obstacles = 2
 
-use_vmas = True  # Set this to True for a great performance speedup
+use_vmas = False  # Set this to True for a great performance speedup
 
 if not use_vmas:
     env = PettingZooEnv(
@@ -775,7 +775,6 @@ def process_batch(batch: TensorDictBase) -> TensorDictBase:
 
 # Training/collection iterations
 for iteration, batch in enumerate(collector):
-    break
     current_frames = batch.numel()
     batch = process_batch(batch)  # Util to expand done keys if needed
     # Loop over groups