Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Gym environments and stable-baselines integration for RL #3215

Merged
merged 15 commits into from
Dec 17, 2020

Conversation

saihv
Copy link
Contributor

@saihv saihv commented Dec 15, 2020

About

OpenAI Gym wrapper for AirSim, and sample drone/car envs for reinforcement learning. Contains DQN examples that will replace the previous hand crafted DQN code. Integrated with stable baselines 3 (PyTorch based) for easy RL training with standard algorithm implementations.

Testing

This PR requires Pytorch to be installed (https://pytorch.org/get-started/locally/), along with the following specific dependencies.

pip install gym tensorboard stable-baselines3

DQN car: Open Neighborhood binary, and run python dqn_car.py
DQN drone: Open LandscapeMountains binary (with powerlines) and run python dqn_drone.py

@saihv saihv marked this pull request as ready for review December 16, 2020 19:36
Copy link
Contributor

@zimmy87 zimmy87 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Tested this locally and it works as expected. Please ignore the two failing Azure pipeline checks, those are failing due to an unrelated issue.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants