hongshaorou/dreamerv3-torch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

added option to parallize

Browse Source
This commit is contained in:
NM512 2023-08-05 22:42:03 +09:00
parent 8571cf656a
commit 02cf57b617
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
configs.yaml
View File

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

2
dreamer.py
View File

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