fchollet / deep-learning-with-python-notebooks

Jupyter notebooks for the code samples of the book "Deep Learning with Python"
MIT License
17.95k stars 8.48k forks source link

TPUs with GAN example on Google colab #220

Open jkylearmstrong opened 1 year ago

jkylearmstrong commented 1 year ago

Does anyone have any insight as to how to update the code on the GAN example https://github.com/fchollet/deep-learning-with-python-notebooks/blob/master/chapter12_part05_gans.ipynb to work with TPUs on Google colab?

It would be really helpful to understand what changes need to be made in order to speed up training