我正在尝试以分布式方式训练深度学习模型。我已经在Keras中建立了模型，现在我需要将模型转换为张量流估计器才能起作用。由于我的数据集已经足够大，并且随着我收集更多的数据而变得越来越大，因此我先使用ImageDataGenerator，然后使用flow_from_directory。我指定我希望批次具有所需形状的大小为32的（299、299、3）。因此，输出为（32、299、299、3）的批次。之前，我只是将对象喂入model.fit_generator，而Keras负责其余的工作。
到目前为止，我已经将Keras模型转换为张量流估计器，但是我无法确定是否可以使用flow_from_directory中的对象来使用tf.train训练数据。我已经尝试编写自己的input_fn，但感到困惑。非常感谢您的帮助。