2019-09-22 11:57:18 +00:00
< img src = "docs/ \_static/img/logo.png" align = "right" width = "40%" />
[](https://travis-ci.com/hill-a/stable-baselines) [](https://stable-baselines.readthedocs.io/en/master/?badge=master)
# Torchy Baselines
PyTorch version of [Stable Baselines ](https://github.com/hill-a/stable-baselines ), a set of improved implementations of reinforcement learning algorithms.
2019-09-22 11:43:01 +00:00
2020-01-22 16:23:42 +00:00
NOTE: Python 3.6 is required!
2019-09-26 14:29:47 +00:00
## Implemented Algorithms
2019-10-25 08:59:15 +00:00
- A2C
2019-09-26 14:29:47 +00:00
- CEM-RL (with TD3)
- PPO
- SAC
- TD3
2019-12-19 14:28:36 +00:00
- SDE support for A2C, PPO, SAC and TD3.
2019-09-26 14:29:47 +00:00
## Roadmap
2019-12-19 14:28:36 +00:00
- cf github Roadmap
2020-01-20 15:19:35 +00:00
2020-01-22 16:23:42 +00:00
## Run the Tests
```
pip install -e .[tests]
make pytest
```
## Type check
```
pip install -e .[tests]
make type
```
## Build the Documentation
```
pip install -e .[docs]
make docs
```
Spell check for the documentation:
```
make spelling
```
2020-01-20 15:19:35 +00:00
## 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}},
}
```
