If you're using Colab and you feel like training your model on GPU is slow, switch to the TPU runtime and tune the "steps_per_execution" parameter in model.compile() (higher = more work being done on device before moving back to host RAM) Can often see a 4-5x speedup.

→ View original post on X — @fchollet,