如何在TPU中的其他模型中使用keras模型

Question

我正在尝试在google colab中将keras模型转换为tpu模型，但是此模型内部还有另一个模型。

看一下代码： https://colab.research.google.com/drive/1EmIrheKnrNYNNHPp0J7EBjw2WjsPXFVJ

这是google tpu文档中示例之一的修改版本： https://colab.research.google.com/github/tensorflow/tpu/blob/master/tools/colab/fashion_mnist.ipynb

如果将sub_model转换并直接使用，它将起作用，但是，如果子模型在另一个模型内部，则它将不起作用。我需要网络的子模型类型，因为我正在尝试训练内部有2个网络（gan = generator + discriminator）的GAN网络，因此，如果此测试可行，那么它也将适用于gan。

我尝试了几件事：

• 转换为模型而不转换子模型，在这种情况下，训练开始时会提示与子模型的输入有关的错误。
• 将模型和子模型都转换为tpu，在这种情况下，转换“父”模型时会提示错误，该异常仅在“层”的末尾显示。
• 仅将子模型转换为tpu，在这种情况下不会提示错误，但训练不会被tpu加速，并且非常慢，就像根本没有转换为tpu一样。
• 是否使用固定的批次大小，两者的结果相同，模型无法正常工作。

有什么想法吗？非常感谢。