Skip to content

Commit 017a266

Browse files
authored
Tuned 2snake config (#217)
1 parent 2a0c38b commit 017a266

File tree

2 files changed

+68
-94
lines changed

2 files changed

+68
-94
lines changed

configs/entity-gym/multisnake_2snakes11len.ron

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,12 @@
1-
// Achieves > 0.98 episodic return: https://wandb.ai/entity-neural-network/enn-ppo/reports/MultiSnake-2-snakes-11-length--VmlldzoxNzgwNzEw
1+
// Achieves ~0.99 episodic return: https://wandb.ai/entity-neural-network/enn-ppo/reports/MultiSnake-snakes-2-len-11--VmlldzoxODE1OTA0
22
ExperimentConfig(
33
version: 0,
44
env: (
55
id: "MultiSnake",
66
kwargs: "{\"num_snakes\": 2, \"max_snake_length\": 11}",
77
),
88
rollout: (
9-
num_envs: 512,
9+
num_envs: 256,
1010
steps: 128,
1111
processes: 16,
1212
),
@@ -23,11 +23,11 @@ ExperimentConfig(
2323
bs: 32768,
2424
lr: 0.005,
2525
max_grad_norm: 10,
26-
micro_bs: 8192,
26+
micro_bs: 4096,
2727
),
2828
ppo: (
29-
ent_coef: 0.03,
30-
gamma: 0.99,
29+
ent_coef: 0.15,
30+
gamma: 0.997,
3131
anneal_entropy: true,
3232
),
3333
)

0 commit comments

Comments
 (0)