search

cjlovering / Towards-Interpretable-Reinforcement-Learning-Using-Attention-Augmented-Agents-Replication

22 stars 6 forks source link

How to run this project #1

Open liuqi8827 opened 4 years ago

liuqi8827 commented 4 years ago

Thanks for your work.

I want to know how can I run this project.

Thanks a lot!

cjlovering commented 4 years ago

Running the python files should just work, although it's not really set up for extensive training. This repo was prototype work. We tried setting up this same model in a distributed setting (linked below). Still, it is difficult to utilize the options offered by the distributed framework. In fact, it takes a ridiculous amount of compute to get decent results. I'd recommend trying to reconfigure the model parameters, etc. for a smaller game.

https://github.com/cjlovering/interpretable-reinforcement-learning-using-attention