From 02cf57b617683a321aad7593d0f6c240387ff732 Mon Sep 17 00:00:00 2001
From: NM512 <morihira3513@gmail.com>
Date: Sat, 5 Aug 2023 22:42:03 +0900
Subject: [PATCH] added option to parallize

---
 configs.yaml | 2 ++
 dreamer.py   | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/configs.yaml b/configs.yaml
index e6dc616..ce77bc1 100644
--- a/configs.yaml
+++ b/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
diff --git a/dreamer.py b/dreamer.py
index aede28c..a02dec3 100644
--- a/dreamer.py
+++ b/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: