Commit 5c899dcc authored by Dipam Chakraborty's avatar Dipam Chakraborty
Browse files

ppg x1 skip 0 npi 18

parent 68e70de1
......@@ -45,8 +45,8 @@ procgen-ppo:
no_done_at_end: False
# Custom switches
skips: 9
n_pi: 9
skips: 0
n_pi: 18
num_retunes: 100
retune_epochs: 6
standardize_rewards: True
......@@ -62,7 +62,7 @@ procgen-ppo:
final_lr: 5.0e-5
lr_schedule: 'linear'
final_entropy_coeff: 0.002
entropy_schedule: True
entropy_schedule: False
# Memory management, if batch size overflow, batch splitting is done to handle it
max_minibatch_size: 2048
......@@ -82,10 +82,10 @@ procgen-ppo:
model:
custom_model: impala_torch_ppg
custom_options:
# depths: [16, 32, 32]
# nlatents: 256
depths: [32, 64, 64]
nlatents: 512
depths: [16, 32, 32]
nlatents: 256
# depths: [32, 64, 64]
# nlatents: 512
init_normed: True
use_layernorm: False
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