AttributeError：“ KerasTPUModel”对象没有属性“ _distribution_strategy”

Question

我正在使用以下设置训练模型：

• Keras模型
• TPU
• Google Colab
• Tensorflow 1.12
• Python3.6

这是我的火车代码：

tpu_model.fit(
    dataset.make_one_shot_iterator(),
    epochs=2,
    steps_per_epoch=100
)

我的数据集是一个简单的 tf.data.Dataset.zip（（图像，标签））。

我得到这个奇怪的错误：

  

AttributeError：“ KerasTPUModel”对象没有属性“ _distribution_strategy”

谢谢。

朱利安