Skip to content

Latest commit

 

History

History
12 lines (8 loc) · 351 Bytes

File metadata and controls

12 lines (8 loc) · 351 Bytes

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

Demo image