Default to multiprocessing vec

thatguy11325 · Oct 15, 2024 · 94bbe51 · 94bbe51
1 parent 5e3f084
commit 94bbe51
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pokemonred_puffer/train.py b/pokemonred_puffer/train.py
@@ -331,6 +331,8 @@ def train(
             vec = pufferlib.vector.Multiprocessing
         elif vec == Vectorization.ray:
             vec = pufferlib.vector.Ray
+        else:
+            vec = pufferlib.vector.Multiprocessing
 
         # TODO: Remove the +1 once the driver env doesn't permanently increase the env id
         env_send_queues = [Queue() for _ in range(2 * config.train.num_envs + 1)]