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 compile(). Seeing a 5-10x speedup is pretty common.
Show this thread
-
So you are saying that a TPU speedup with minimal code changes is possible in tf.Keras?
This is the setup needed to start running on TPU on Colab: https://keras.io/examples/vision/xray_classification_with_tpus/#introduction--setup … Besides this, Keras provides you with a way to reach full device utilization (steps_per_execution), which is a necessary condition for TPU training to provide a large speedup.
-
-
Thanks man your awesome !
Thanks. Twitter will use this to make your timeline better. UndoUndo
-
-
-
This is great...will try this as soon as Google restores my free GPU , TPU runtime quota for the week
Thanks. Twitter will use this to make your timeline better. UndoUndo
-
Loading seems to be taking a while.
Twitter may be over capacity or experiencing a momentary hiccup. Try again or visit Twitter Status for more information.