python-3.x - AttributeError：“KerasTPUModel”对象没有属性“_ckpt_saved_epoch”

Question

我正在用 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))

错误：

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

提前感谢您的帮助

标签： python-3.xtf.keras