我正在用google colab TPU训练一个keras模型。我的代码以前在CPU和GPU上运行得很成功。但是，当我把代码改到TPU版本时，我遇到了一些错误。
下面是我的转换代码：

model = tf.contrib.tpu.keras_to_tpu_model(
model,
strategy=tf.contrib.tpu.TPUDistributionStrategy(
    tf.contrib.cluster_resolver.TPUClusterResolver(TPU_WORKER)))

下面是我的编译代码：

model.compile(optimizer=tf.train.AdamOptimizer(learning_rate=1e-3), 
loss='categorical_crossentropy', metrics=['accuracy'])

运行以下代码时发生错误：

model.fit(X_train,y_train,epochs=10,batch_size=64*8,validation_data=(X_test,y_test))

错误：
属性错误：“KerasTPUModel”对象没有属性“_检查点保存的纪元”
谢谢你的帮助