Skip to content

thimabru1010/Flappy-Bird-Double-DQN-Pytorch

 
 

Repository files navigation

Flappy-Bird-Double-DQN-Pytorch

Train an agent to play flappy bird game using double DQN model and implement it with pytorch.

Result

Installations (suggest using virtualenv)

  • Pygame
$ pip install pygame
  • ple
$ git clone https://github.com/ntasfi/PyGame-Learning-Environment.git
$ cd PyGame-Learning-Environment/
$ pip install -e .
  • gym
$ pip install gym
  • gym-ple
$ pip install gym-ple

How to run

  • Train
# If cuda is available, add --cuda Y.
# Add --ckpt [ckpt_file] to train from this checkpoint. 
$ python main.py --mode train 
  • Evaluation
# If cuda is available, add --cuda Y.
$ python main.py --mode eval --ckpt [checkpoint.pth.tar]

About

Train an agent to play flappy bird game using double DQN model and implement it with pytorch.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 95.8%
  • Makefile 3.5%
  • Dockerfile 0.7%