Skip to content

Commit 6e803d9

Browse files
authored
Merge pull request #275 from leonardo-toffalini/asteroids
adds sweeped hparams to asteroids config
2 parents f0f952a + 05af323 commit 6e803d9

File tree

1 file changed

+23
-2
lines changed

1 file changed

+23
-2
lines changed

pufferlib/config/ocean/asteroids.ini

Lines changed: 23 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,5 +12,26 @@ num_envs = 1024
1212
size = 500
1313

1414
[train]
15-
total_timesteps = 150_000_000
16-
minibatch_size = 32768
15+
adam_beta1 = 0.975493290069733
16+
adam_beta2 = 0.9999436458974764
17+
adam_eps = 6.915036275112011e-08
18+
anneal_lr = true
19+
batch_size = auto
20+
bptt_horizon = 64
21+
checkpoint_interval = 200
22+
clip_coef = 0.18588778503512546
23+
ent_coef = 0.0016620361911332262
24+
gae_lambda = 0.8400278040617952
25+
gamma = 0.9998708818940873
26+
learning_rate = 0.00502237062536979
27+
max_grad_norm = 0.7306435358436453
28+
max_minibatch_size = 32768
29+
minibatch_size = 8192
30+
prio_alpha = 0.9165093859993415
31+
prio_beta0 = 0.8869674411376214
32+
total_timesteps = 100_000_000
33+
update_epochs = 1
34+
vf_clip_coef = 0.1
35+
vf_coef = 2.960148388519086
36+
vtrace_c_clip = 1.0767718761515104
37+
vtrace_rho_clip = 4.132507367126342

0 commit comments

Comments
 (0)