Keras可以预取像tensorflow Dataset这样的数据吗？

Question

在TensorFlow的Dataset API中，我们可以用来dataset.prefetch(buffer_size=xxx)在GPU处理当前批次的数据时预加载其他批次的数据，因此，我可以充分利用GPU。

我将使用Keras，想知道是否keras有类似的API让我充分利用GPU，而不是串行执行：读取批处理0->处理批处理0->读取批处理1->处理批处理1->。 ..

我简要浏览了一下kerasAPI，但没有看到有关预取的描述。

Answer 1

如果调用fit_generator与workers> 1，USE_ multiprocessing=True，它会预取的queue_size批次。

来自文档：max_queue_size：整数。生成器队列的最大大小。如果未指定，max_queue_size 将默认为 10。