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

set model size L for minecraft in config

Browse Source
This commit is contained in:
NM512 2023-07-24 21:42:04 +09:00
parent 9b97035310
commit 22e9f8de7b
1 changed files with 19 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
configs.yaml
View File

@ -161,25 +161,37 @@ atari100k:
steps: 4e5 steps: 4e5
envs: 1 envs: 1
action_repeat: 4 action_repeat: 4
train_ratio: 1024
video_pred_log: true
eval_episode_num: 100 eval_episode_num: 100
actor_dist: 'onehot'
imag_gradient: 'reinforce'
stickey: False stickey: False
lives: unused lives: unused
noops: 30 noops: 30
resize: opencv resize: opencv
actions: needed actions: needed
actor_dist: 'onehot'
train_ratio: 1024
imag_gradient: 'reinforce'
time_limit: 108000 time_limit: 108000
minecraft: minecraft:
task: minecraft_diamond task: minecraft_diamond
break_speed: 100.0 step: 1e8
envs: 16 envs: 16
action_repeat: 1
train_ratio: 16 train_ratio: 16
log_keys_max: '^log_inventory.*' video_pred_log: true
encoder: {mlp_keys: 'inventory|inventory_max|equipped|health|hunger|breath|reward', cnn_keys: 'image'} dyn_hidden: 1024
decoder: {mlp_keys: 'inventory|inventory_max|equipped|health|hunger|breath', cnn_keys: 'image'} dyn_deter: 4096
units: 1024
reward_layers: 5
cont_layers: 5
value_layers: 5
actor_layers: 5
encoder: {mlp_keys: 'inventory|inventory_max|equipped|health|hunger|breath|reward', cnn_keys: 'image', cnn_depth: 96, mlp_layers: 5, mlp_units: 1024}
decoder: {mlp_keys: 'inventory|inventory_max|equipped|health|hunger|breath', cnn_keys: 'image', cnn_depth: 96, mlp_layers: 5, mlp_units: 1024}
actor_dist: 'onehot'
imag_gradient: 'reinforce'
break_speed: 100.0
time_limit: 36000 time_limit: 36000
debug: debug: