Bezos is a light Deep RL framework that you can fork and extend.
Report Bug
·
Request Feature
There are many great reinforcement learning frameworks on GitHub, however, I didn't find one that was light and small enough to let me fork it and add my own algorithms while still feeling in control. I thus created Bezos, a very light RL framework which implements all the boring things so that you don't have to.
- Rollout class (supports GAE, reward to go, and recurrent neural network)
- Runner class: loads a Gym env and runs one or multiple agents on the environment to generate rollouts
- A simple Actor Critic Network
- RL Algorithm: PPO Paper on Arxiv
- RL Algorithm: A2C OpenAI blog post
- Supports all the OpenAI Gym env
- I added all the Vizdoom envs as well (look into kits/)
- Supports the marLo env Github (look into kits/)
- A bunch of useful OpenAI Gym wrappers. Includes frame skipping, RGB→Grayscale, cropping, and many more. (look into envs.py)
The framework uses Pytorch to do all the deep learning stuff, numpy, and OpenAI Gym
Bezos has been designed to be configured before each run with a YAML file. Some examples of those YAML config files can be found in the configs folder. If you want to know what a parameter does (they are all self explanatory) check the source :)
- Create a new conda env (yo don't use python 2 ok)
pip install -r requirements.txt
python bezos.py --config ./configs/ppo-minecraft.yaml train
python bezos.py --config ./configs/ppo-minecraft.yaml evaluate --det
Contributions are what make the open source community such an amazing place to be learn, inspire, and create. Any contributions you make are greatly appreciated.
- Fork the Project
- Create your Feature Branch (
git checkout -b feature/AmazingFeature
) - Commit your Changes (
git commit -m 'Add some AmazingFeature
) - Push to the Branch (
git push origin feature/AmazingFeature
) - Open a Pull Request
Distributed under the MIT License. See LICENSE
for more information.
Justin Glibert - justin@glibert.io