PyTorch implementation of DQN.

breakout.mp4

pong.mp4

spaceinvaders-defense.mp4

stargunner.mp4

Setup

Run the following commands:

apt-get update
apt-get install git wget cmake python3-opencv ffmpeg -y

Create a virtual environment and install requirements:

cd dqn
pip install -r requirements.txt

Training

Run:

cd dqn
python run.py --env BreakoutNoFrameskip-v4 --config rainbow

where env is a gym environment name, and config is the name of a file in the "configs" directory (minus the .gin extension).

Evaluation

Run:

python generate_video.py --env BreakoutNoFrameskip-v4 --run gcp-breakout

where run is the name of a directory with agent checkpoints.