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

added benchmark task Crafter

Browse Source
This commit is contained in:
NM512
2023-06-18 00:02:22 +09:00
parent 9c58ab62c0
commit 5dce8cf13b
6 changed files with 111 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
configs.yaml
View File

@@ -138,6 +138,25 @@ dmc_vision:
encoder: {mlp_keys: '$^', cnn_keys: 'image'}
decoder: {mlp_keys: '$^', cnn_keys: 'image'}
crafter:
task: crafter_reward
step: 1e6
action_repeat: 1
envs: 1
train_ratio: 512
video_pred_log: true
dyn_hidden: 1024
dyn_deter: 4096
units: 1024
reward_layers: 5
cont_layers: 5
value_layers: 5
actor_layers: 5
encoder: {mlp_keys: '$^', cnn_keys: 'image', cnn_depth: 96, mlp_layers: 5, mlp_units: 1024}
decoder: {mlp_keys: '$^', cnn_keys: 'image', cnn_depth: 96, mlp_layers: 5, mlp_units: 1024}
actor_dist: 'onehot'
imag_gradient: 'reinforce'
atari100k:
steps: 4e5
envs: 1