Commit 62e9c9bc authored by Dipam Chakraborty's avatar Dipam Chakraborty
Browse files

4x model

parent 04d5264d
...@@ -45,9 +45,9 @@ procgen-ppo: ...@@ -45,9 +45,9 @@ procgen-ppo:
no_done_at_end: False no_done_at_end: False
# Custom switches # Custom switches
skips: 2 skips: 0
n_pi: 16 n_pi: 16
num_retunes: 14 num_retunes: 16
retune_epochs: 6 retune_epochs: 6
standardize_rewards: True standardize_rewards: True
aux_mbsize: 4 aux_mbsize: 4
...@@ -88,8 +88,10 @@ procgen-ppo: ...@@ -88,8 +88,10 @@ procgen-ppo:
model: model:
custom_model: impala_torch_ppg custom_model: impala_torch_ppg
custom_model_config: custom_model_config:
depths: [32, 64, 64] # depths: [32, 64, 64]
nlatents: 512 # nlatents: 512
depths: [64, 128, 128]
nlatents: 1024
init_normed: True init_normed: True
use_layernorm: False use_layernorm: False
diff_framestack: True diff_framestack: True
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment