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

set "envs" according to the paper

Browse Source
This commit is contained in:
NM512 2023-06-17 15:37:34 +09:00
parent ccc98d46c9
commit 784a69f80c
1 changed files with 12 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
configs.yaml
View File

@ -120,22 +120,27 @@ defaults:
disag_units: 400 disag_units: 400
disag_action_cond: False disag_action_cond: False
dmc_vision:
steps: 1e6
train_ratio: 512
video_pred_log: true
encoder: {mlp_keys: '$^', cnn_keys: 'image'}
decoder: {mlp_keys: '$^', cnn_keys: 'image'}
dmc_proprio: dmc_proprio:
steps: 5e5 steps: 5e5
action_repeat: 2
envs: 4
train_ratio: 512 train_ratio: 512
video_pred_log: false video_pred_log: false
encoder: {mlp_keys: '.*', cnn_keys: '$^'} encoder: {mlp_keys: '.*', cnn_keys: '$^'}
decoder: {mlp_keys: '.*', cnn_keys: '$^'} decoder: {mlp_keys: '.*', cnn_keys: '$^'}
dmc_vision:
steps: 1e6
action_repeat: 2
envs: 4
train_ratio: 512
video_pred_log: true
encoder: {mlp_keys: '$^', cnn_keys: 'image'}
decoder: {mlp_keys: '$^', cnn_keys: 'image'}
atari100k: atari100k:
steps: 4e5 steps: 4e5
envs: 1
action_repeat: 4 action_repeat: 4
eval_episode_num: 100 eval_episode_num: 100
stickey: False stickey: False