/Renforcement-Learning-Tetris-DQN

Deep reinforcment learning implementation on Tetris Atari game

Primary LanguagePythonApache License 2.0Apache-2.0

Renforcement-Learning-Tetris-DQN

Deep reinforcment learning implementation on Tetris Atari game with PyTorch.

To train the model simply run the train.py file and to test it run the test.py

Check the after training video to see how crazy the results are.