如何使用Google Cloud TPU培训keras模型

C.L*_*Lee 9 google-cloud-platform keras

从理论上讲,我们能够在任何支持Tensorflow的平台上训练Tensorflow支持的Keras 模型.但是,我似乎无法在Google的文档https://cloud.google.com/tpu/docs/中找到有关如何操作的任何信息.

小智 0

我们在 TPU 存储库中有一个示例: https://github.com/tensorflow/tpu/blob/master/models/experimental/cifar_keras/cifar_keras.py

  • 谢谢,看来这个例子中仍然有相当多的 Tensorflow 内容。例如损失函数和优化器。此外,在示例中,Keras 是从“tf.contrib”导入的。我知道 Keras 是由 Google 团队开发的,但这是现在使用 Keras 的推荐方式吗? (2认同)