Skip to content

Commit f2ae794

Browse files
authored
Merge pull request #276 from y-hesse/3.0
sweeped params and policy for 2048
2 parents 6e803d9 + 69a7968 commit f2ae794

File tree

3 files changed

+250
-158
lines changed

3 files changed

+250
-158
lines changed

pufferlib/config/ocean/g2048.ini

Lines changed: 30 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,9 +4,36 @@ env_name = puffer_g2048
44
policy_name = G2048
55
rnn_name = Recurrent
66

7+
[policy]
8+
hidden_size = 256
9+
10+
[rnn]
11+
input_size = 256
12+
hidden_size = 256
13+
14+
[vec]
15+
num_envs = 4
16+
717
[env]
8-
num_envs = 4096
18+
num_envs = 4024
919

1020
[train]
11-
total_timesteps = 1_000_000_000
12-
minibatch_size = 32768
21+
total_timesteps = 600_000_000
22+
adam_beta1 = 0.9529488439604378
23+
adam_beta2 = 0.9993901829477296
24+
adam_eps = 2.745365927413118e-7
25+
bptt_horizon = 64
26+
clip_coef = 0.596573170393339
27+
ent_coef = 0.02107417730003862
28+
gae_lambda = 0.9940613415815854
29+
gamma = 0.9889857974154952
30+
learning_rate = 0.0032402460796988127
31+
max_grad_norm = 1.0752406726589745
32+
minibatch_size = 16384
33+
prio_alpha = 0.25297099593586336
34+
prio_beta0 = 0.940606268942572
35+
vf_clip_coef = 0.1
36+
vf_coef = 1.6362878279900643
37+
vtrace_c_clip = 0
38+
vtrace_rho_clip = 1.2917509971869054
39+
anneal_lr = False

0 commit comments

Comments
 (0)