1.只能使用tensorflow的keras接口

2.數據的維度或者batch_size必須是128的倍數，

這樣TPU才能發揮最大性能

3.目前不支持keras的優化器，而是需要像下面這樣來書寫

optimizer=tf.train.AdamOptimizer()

４.在代碼訓練前，我們需要使用keras_to_tpu_model來轉化模型.

?

?

總結

以上是生活随笔為你收集整理的使用TPU的注意事项的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。