google colab TPU上自定义图像数据生成器（带增强）的示例（keras）

我用Keras编写了一个基于resnet的模型，对图像数据进行增强训练。我打算使用google colab TPU，但我能找到的所有示例都不起作用，有人能给我举个例子吗？

你检查过这个示例吗

笔记本有一些增强功能（

data\u augment

方法），要使用一些resnet，您可以用

tf.keras.applications.ResNet50

替换模型