gpt4 book ai didi

tensorflow-datasets - tf.dataset.prefetch() buffer_size 含义

转载 作者:行者123 更新时间:2023-12-04 15:17:38 29 4
gpt4 key购买 nike

dataset = dataset.batch(50)
dataset = dataset.prefetch(buffer_size=1)

是预取 1 个批次还是 1 个元素?

根据 tensorflow 中的 API 文档,buffer_size 是元素预取的最大数量。但它似乎是批处理数据集后的批次数。

最佳答案

由于您使用的是 dataset.prefetch(buffer_size=1)之后 dataset.batch() , 表示 它将预取 1 批 .

关于tensorflow-datasets - tf.dataset.prefetch() buffer_size 含义,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/51643268/

29 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com