且构网

分享程序员开发的那些事...
且构网 - 分享程序员编程开发的那些事

Keras 可以预取 tensorflow Dataset 之类的数据吗?

更新时间:2023-02-26 08:00:54

如果调用 fit_generatorworkers > 1,使用_multiprocessing=True,它将预取 queue_size 个批次.

If you call fit_generator with workers > 1, use_multiprocessing=True, it will prefetch queue_size batches.

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

From docs: max_queue_size: Integer. Maximum size for the generator queue. If unspecified, max_queue_size will default to 10.