zblhero/pytorch-dqn

Deep Q-Learning Network in pytorch (not actively maintained)

pytoch-dqn

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

Credit

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

Requirements

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

Usage

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