Chrispresso / SuperMarioBros-AI

332 stars 69 forks source link

How can i enable GPU when training CNN for new population? #11

Open homata123 opened 2 years ago

homata123 commented 2 years ago

Your code is simply perfect and helpful. But there is an issue when training new populations, it is the training speed. I didn't see it using GPU when training, and the speed becomes lower when it comes to big gens, etc from gen 500. So can you manage the code for enabling GPU during the process in a new version of pull request? Thank you for your work one more time.