Skip to content

Latest commit

 

History

History

02-double-dqn

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Double Deep Q-Learning Network (Double-DQN)

This is a pytorch implementation of "Deep Reinforcement Learning with Double Q-learning". In this lastest version, I use the Atari games. But in the future, I will bring the Flappy Bird back!

Requirements

  • python-3.5.2
  • openai-gym
  • pytorch-0.4.0
  • opencv-python

Installation

Install OpenAI Baselines (the openai-baselines update so quickly, please use the older version as blow, will solve in the future.)

# clone the openai baselines
git clone https://github.com/openai/baselines.git
cd baselines
git checkout 366f486
pip install -e .

Instruction to run the code

Train the Network:

python train_network.py --cuda (if you have a GPU, you can use this flag)

Test the Network:

python demo.py

Download the Pre-trained Model

Please download them from the Google Driver, then put the saved_models under the current folder.

Results

Training Performance

results

Demo: SpaceInvadersNoFrameskip-v4

demo