This sample is published as part of the corresponding blog article at https://www.toptal.com/deep-learning/pytorch-reinforcement-learning-tutorial
Visit https://www.toptal.com/blog and subscribe to our newsletter to read great posts!
pip3 install -r requirements.txtto install dependencies.
python dqn.py testto run pretrained neural network model.
python dqn.py trainto train the model from the beginning. You can also increase FPS in game/flappy_bird.py script for faster training.