added option to parallize

2023-08-05 22:42:03 +09:00
parent 8571cf656a
commit 02cf57b617
2 changed files with 3 additions and 1 deletions
--- 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: