Tensorflow TPU踩坑心得

Hai Feng Kao
1 min readJul 14, 2020
  1. 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都要有明確而且固定的大小

--

--