stable-baselines3/README.md

Torchy Baselines

PyTorch version of Stable Baselines, a set of improved implementations of reinforcement learning algorithms.

Implemented Algorithms

• A2C

• CEM-RL (with TD3)

• PPO

• SAC

• TD3

• SDE support for A2C, PPO, SAC and TD3.

Roadmap

• cf github Roadmap

Citing the Project

To cite this repository in publications:

@misc{torchy-baselines,
  author = {Raffin, Antonin and Dormann, Noah and Hill, Ashley and Ernestus, Maximilian and Gleave, Adam and Kanervisto, Anssi},
  title = {Torchy Baselines},
  year = {2019},
  publisher = {GitHub},
  journal = {GitHub repository},
  howpublished = {\url{https://github.com/araffin/torchy-baselines}},
}