Fatal编程技术网
  • tensorflow/
  • Tensorflow GCP TPU上的罐装模型

Tensorflow GCP TPU上的罐装模型

tensorflow google-cloud-platform deep-learning

Tensorflow GCP TPU上的罐装模型,tensorflow,google-cloud-platform,deep-learning,google-cloud-tpu,Tensorflow,Google Cloud Platform,Deep Learning,Google Cloud Tpu,Google的TPU要求您将tensorflow估计器移植到TPUEstimators，但我似乎不知道“固定”估计器（如DNNClassifier）需要做哪些更改-似乎只有输入函数真正需要更改才能使用tf.data.Dataset，但我可能错了。感谢您的帮助。TPU罐装估计器正在研制中。但是，要回答你的问题，我相信你必须 需要改变的事情 确保优化器包装在tf.contrib.tpu.CrossShardOptimizer中 tf.ops摘要需要移除或替换为NOOP 让估计器的模型返回TPUEs

Google的TPU要求您将tensorflow估计器移植到TPUEstimators，但我似乎不知道“固定”估计器（如
DNNClassifier
）需要做哪些更改-似乎只有输入函数真正需要更改才能使用
tf.data.Dataset
，但我可能错了。感谢您的帮助。

TPU罐装估计器正在研制中。但是，要回答你的问题，我相信你必须

需要改变的事情

  • 确保优化器包装在tf.contrib.tpu.CrossShardOptimizer中
  • tf.ops摘要需要移除或替换为NOOP
  • 让估计器的模型返回TPUEstimatorSpec
    • 执行静态批大小（XLA需要我相信）< /LI>之类的事情
可能还有其他一些事情