Tensorflow 如何在带有ImageDataGenerator的Kaggle/Colab上使用TPU？

我有一个目录和数据框，如下所示：

|---classA
|-----1.jpg
|-----2.jpg
|---classB
|-----23.jpg
|-----124.jpg

dataframe由两列组成：图像的名称及其类

name  | class
1.jpg |   1
2.jpg |   1
23.jpg|   2

---

我使用并使用方法

flow\u from\u directory

。我目前正在使用GPU进行训练，但因为它很大，需要很多时间，所以我想切换到TPU。我搜索了Kaggle和Colab的TPU指南，他们使用TFRecord读取数据。有谁能帮助我如何使用ImageDataGenerator中的TFRecord吗？

因此我发现ImageDataGenerator不能在TPU上使用。您必须改用tf.data.Dataset。

AFAIK，您不能在TPU上使用ImageGens！是的，我刚在TPU上找到Kaggle员工关于IDG的评论。我是否可以以某种方式“复制”IDG？