即使在 Keras 1.2.2 版本中,参考合并,虽然已经包含了多处理功能,但由于磁盘读取速度的限制,model.fit_generator()
的速度仍然比 model.fit()
慢大约 4-5 倍。要如何通过额外的多处理来加速呢?
回答:
你可以查看 fit_generator()
的 workers
和 max_queue_size
参数,具体可以在文档中找到。增加 workers
的数量可以创建更多的线程来加载数据到队列中,从而为你的网络提供数据。不过,填充队列可能会导致内存问题,因此你可能需要减少 max_queue_size
来避免这种情况。