Python 3.x 使用TensorFlow 2.0 keras:NotFoundError FetchOutputs节点：未找到[Op:AutoSharedDataSet]

我试图利用谷歌Colab上的TPU。我已经跟着导游走了 详情如下:

resolver = tf.distribute.cluster_resolver.TPUClusterResolver(tpu='grpc://' + os.environ['COLAB_TPU_ADDR'])
tf.config.experimental_connect_to_host(resolver.master())
tf.tpu.experimental.initialize_tpu_system(resolver)
strategy = tf.distribute.experimental.TPUStrategy(resolver)

当我运行

superres.fit（数据集，epochs=10）

时，我得到以下错误

NotFoundError: FetchOutputs node : not found [Op:AutoShardDataset]

请帮忙

---

链接到colab:

Hi@Shashikant-Kadam，您能添加数据集示例吗？或者提供一个可共享的链接？

with strategy.scope():
  superres = SuperResolution((44, 64, 64))
  superres.compile(loss = tf.keras.losses.MeanSquaredError(), optimizer = tf.keras.optimizers.Adam())

dataset = tf.data.Dataset.from_tensor_slices((Dataset64.reshape(len(Dataset64), 44, 64, 64, 1), Dataset128.reshape(len(Dataset128), 88, 128, 128, 1)))

dataset = dataset.shuffle(10).batch(1)
superres.fit(dataset, epochs = 10)

NotFoundError: FetchOutputs node : not found [Op:AutoShardDataset]