search

araffin / rl-tutorial-jnrr19

Stable-Baselines tutorial for Journées Nationales de la Recherche en Robotique 2019
MIT License
591 stars 113 forks source link

How to do hyper parameter tuning for SB3 algorithm? #21

Open wbzhang233 opened 7 months ago

wbzhang233 commented 7 months ago

How to do hyper parameter tuning for SB3 algorithm such as PPO, A2C, DQN?

araffin commented 7 months ago

Hello, you can have a look at the "automatic hyperparameter tuning" slides/video and notebook from https://araffin.github.io/tools-for-robotic-rl-icra2022/

See https://colab.research.google.com/github/araffin/tools-for-robotic-rl-icra2022/blob/main/notebooks/optuna_lab.ipynb

you will to slightly update the code to use gymnasium instead of gym.