How to use the prioritized code to run atari games?

MorvanZhou / Reinforcement-learning-with-tensorflow

Simple Reinforcement learning tutorials, 莫烦Python 中文AI教学

https://mofanpy.com/tutorials/machine-learning/reinforcement-learning/

MIT License

8.92k stars 5.01k forks source link

How to use the prioritized code to run atari games? #1

Closed shanyou92 closed 7 years ago

shanyou92 commented 7 years ago

Hi, I want to run atari games based on the prioritized experience replay. However, the current code does not work. Any ideas?

MorvanZhou commented 7 years ago

This code only works on a 1D state representation (state number,), I think atari will output a 2D state(img_height, img_width). If this is the case, please modify the relevant parts accordingly.