mindspore.dataset.config.set_prefetch_size
- mindspore.dataset.config.get_prefetch_size()
获取数据处理管道的输出缓存队列长度。 如果 set_prefetch_size 方法未被调用,那么将会返回默认值16。
- 返回:
int,表示预取的总行数。
样例:
>>> # Get the global configuration of prefetch size. >>> # If set_prefetch_size() is never called before, the default value(16) will be returned. >>> prefetch_size = ds.config.get_prefetch_size()