search

werner-duvaud / muzero-general

MuZero
https://github.com/werner-duvaud/muzero-general/wiki/MuZero-Documentation
MIT License
2.46k stars 606 forks source link

Using TPUs #132

Open StrangeTcy opened 3 years ago

StrangeTcy commented 3 years ago

So long as you're offering a notebook that can be run in Colab anyway, why don't you expand the code and include an option of running training (and then possibly influence) on TPUs?

ahainaut commented 3 years ago

Hi, Indeed, It would be nice to include this option but this will probably requires modifying a certain amount of code for compatibility. We do not plan to integrate this feature for now but contributions are welcomed !