added option to parallize

configs.yaml

@@ -7,6 +7,7 @@ defaults:
   offline_evaldir: ''
   seed: 0
   steps: 1e6
+  parallel: False
   eval_every: 1e4
   eval_episode_num: 10
   log_every: 1e4
@@ -176,6 +177,7 @@ atari100k:
 minecraft:
   task: minecraft_diamond
   step: 1e8
+  parallel: True
   envs: 16
   # no eval
   eval_episode_num: 0

dreamer.py

@@ -266,7 +266,7 @@ def main(config):
     make = lambda mode: make_env(config, mode)
     train_envs = [make("train") for _ in range(config.envs)]
     eval_envs = [make("eval") for _ in range(config.envs)]
-    if config.envs > 1:
+    if config.parallel:
         train_envs = [Parallel(env, "process") for env in train_envs]
         eval_envs = [Parallel(env, "process") for env in eval_envs]
     else: