RL Baselines3 Zoo Docs - A Training Framework for Stable Baselines3 — RL Baselines3 Zoo 2.6.1a0 documentation (original) (raw)
RL Baselines3 Zoo s a training framework for Reinforcement Learning (RL), using Stable Baselines3 (SB3), reliable implementations of reinforcement learning algorithms in PyTorch.
Github repository: https://github.com/DLR-RM/rl-baselines3-zoo
It provides scripts for training, evaluating agents, tuning hyperparameters, plotting results and recording videos.
In addition, it includes a collection of tuned hyperparameters for common environments and RL algorithms, and agents trained with those settings.
User Guide
- Installation
- Getting Started
- Train an Agent
- Plot Scripts
- Enjoy a Trained Agent
- Custom Environment
- Configuration
- Integrations
- Hyperparameter Tuning
- Stable Baselines Jax (SBX)
Citing RL Baselines3 Zoo
To cite this project in publications:
@misc{rl-zoo3, author = {Raffin, Antonin}, title = {RL Baselines3 Zoo}, year = {2020}, publisher = {GitHub}, journal = {GitHub repository}, howpublished = {\url{https://github.com/DLR-RM/rl-baselines3-zoo}}, }
Contributing
To any interested in making the rl baselines better, there are still some improvements that need to be done. You can check issues in the repo.
If you want to contribute, please read CONTRIBUTING.md first.