Bart/bullet3
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Update buffer size

Browse Source
This commit is contained in:
Antonin RAFFIN
2020-01-02 14:47:30 +01:00
parent 59c61a46bb
commit adad4dc402
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/pybullet/gym/pybullet_envs/stable_baselines/train.py
View File

@@ -45,7 +45,7 @@ if __name__ == '__main__':
# Tuned hyperparameters from https://github.com/araffin/rl-baselines-zoo
hyperparams = {
'sac': dict(batch_size=256, gamma=0.98, policy_kwargs=dict(layers=[256, 256]),
learning_starts=10000, buffer_size=int(1e6), tau=0.01),
learning_starts=10000, buffer_size=int(2e5), tau=0.01),
'td3': dict(batch_size=100, policy_kwargs=dict(layers=[400, 300]),
learning_rate=1e-3, learning_starts=10000, buffer_size=int(1e6),