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.txt
to install dependencies.python dqn.py test
to run pretrained neural network model.python dqn.py train
to train the model from the beginning. You can also increase FPS in game/flappy_bird.py script for faster training.References: