Tetris DQN with PyGame and PyTorch

Description

This my source code for an effective Tetris DQN agent which plays the game in real-time.

Neural Network Model

The neural network model was trained in main.py, and tested in test_nn.py.

The model can be found under model/checkpoint.tar.

Demo