Tensorflow TPU踩坑心得
1 min readJul 14, 2020
- Google有提供一個簡單的TPU使用範例
Predict Shakespeare with Cloud TPUs and Keras — Show …colab.research.google.com › tools › colab
2. 不能使用tf.py_function。不管是在build model還是preprocess dataset,都不能使用tf.py_function去執行python的函數
3. 在TPU上使用tensorflow的函數,會有特殊的限制,請參考
4. 別想tf.RaggedTensor了,忘了它吧。所有的tensor都要有明確而且固定的大小