Awesome Open Source
Awesome Open Source

[PYTORCH] Deep Q-learning for playing Flappy Bird

Introduction

Here is my python source code for training an agent to play flappy bird. It could be seen as a very basic example of Reinforcement Learning's application.


Result

How to use my code

With my code, you can:

  • Train your model from scratch by running python train.py
  • Test your trained model by running python test.py

Trained models

You could find my trained model at trained_models/flappy_bird

Requirements

  • python 3.6
  • pygame
  • cv2
  • pytorch
  • numpy


Alternative Project Comparisons
Related Awesome Lists
Top Programming Languages

Get A Weekly Email With Trending Projects For These Topics
No Spam. Unsubscribe easily at any time.
Python (889,550
Learning (75,673
Deep (39,305
Pytorch (22,629
Reinforcement Learning (4,870
Pygame (3,626
Flappy Bird (493
Deep Q Network (113
Deep Q Learning (95