Skip to content

Commit 309ad8c

Browse files
authored
[ci skip] Tune RecurrentPPO Swimmer-v3 (#360)
1 parent fb9383a commit 309ad8c

File tree

2 files changed

+6
-1
lines changed

2 files changed

+6
-1
lines changed

CHANGELOG.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
- Upgraded to SB3 >= 1.8.0
55

66
### New Features
7+
- Tuned hyperparameters for RecurrentPPO on Swimmer
78

89
### Bug fixes
910
- Set ``highway-env`` version to 1.5 and ``setuptools to`` v65.5 for the CI

hyperparams/ppo_lstm.yml

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -339,7 +339,11 @@ Ant-v3: &mujoco-defaults
339339
Swimmer-v3:
340340
<<: *mujoco-defaults
341341
gamma: 0.9999
342-
342+
n_envs: 4
343+
n_steps: 1024
344+
batch_size: 256
345+
learning_rate: !!float 6e-4
346+
gae_lambda: 0.98
343347

344348
# 10 mujoco envs
345349

0 commit comments

Comments
 (0)