使用TPU的注意事项

1.只能使用tensorflow的keras接口

2.数据的维度或者batch_size必须是128的倍数，

这样TPU才能发挥最大性能

3.目前不支持keras的优化器，而是需要像下面这样来书写

optimizer=tf.train.AdamOptimizer()

４.在代码训练前，我们需要使用keras_to_tpu_model来转化模型.

 

 

总结

以上是生活随笔为你收集整理的使用TPU的注意事项的全部内容，希望文章能够帮你解决所遇到的问题。

如果觉得生活随笔网站内容还不错，欢迎将生活随笔推荐给好友。