Skip to content

Commit 9f56612

Browse files
author
Joseph Suarez
committed
nle basic
1 parent c3a24f5 commit 9f56612

File tree

3 files changed

+670
-11
lines changed

3 files changed

+670
-11
lines changed

pufferlib/config/nethack.ini

Lines changed: 22 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,31 @@
11
[base]
22
package = nethack
33
env_name = nethack
4+
policy_name = Policy
5+
rnn_name = Recurrent
46

57
[vec]
6-
num_envs = 128
8+
num_envs = 8192
79
num_workers = 16
8-
batch_size = 64
10+
batch_size = 4096
911

1012
[train]
11-
batch_size = 8192
12-
minibatch_size = 2048
13-
update_epochs = 1
13+
total_timesteps = 90_000_000
14+
adam_beta1 = 0.8946507418260217
15+
adam_beta2 = 0.9
16+
adam_eps = 0.0001
17+
batch_size = auto
1418
bptt_horizon = 64
15-
total_timesteps = 10_000_000
16-
anneal_lr = False
19+
clip_coef = 0.19696765958267629
20+
ent_coef = 0.0005690816545012474
21+
gae_lambda = 0.747650023961198
22+
gamma = 0.9997053654668936
23+
learning_rate = 0.044482546441415506
24+
max_grad_norm = 2.2356112188495723
25+
minibatch_size = 32768
26+
prio_alpha = 0.98967001208896
27+
prio_beta0 = 0.09999999999999998
28+
vf_clip_coef = 2.178492167689251
29+
vf_coef = 1.6832989594296321
30+
vtrace_c_clip = 2.878171091654008
31+
vtrace_rho_clip = 0.7876748061547312

0 commit comments

Comments
 (0)