This project is pytorch implementation of Human-level control through deep reinforcement learning and I also plan to implement the following ones:

• Prioritized Experience Replay
• Deep Reinforcement Learning with Double Q-learning

This project reuses most of the code in https://github.com/berkeleydeeprlcourse/homework/tree/master/hw3

• python 3.5
• gym (built from source)
• pytorch (built from source)

To train a model:

$ python main.py
# To train the model using ram not raw images, helpful for testing
$ python ram.py

The model is defined in dqn_model.py

The algorithm is defined in dqn_learn.py

The running script and hyper-parameters are defined in main.py