Tensorflow tf.contrib.data.prefetch_至_设备_Tensorflow

Tensorflow tf.contrib.data.prefetch_至_设备

tensorflow

Tensorflow tf.contrib.data.prefetch_至_设备,tensorflow,Tensorflow,由于一些成本函数（如tf.nn.ctc_loss）在CPU上运行，因此不需要GPU内存上的标签，我们如何在小批量中只预取训练数据而不预取标签。您需要创建两个数据集对象才能实现这一点。拥有两个数据集有很多用例，所以这并不异常请注意，尽管预回迁不会预回迁到GPU，但这似乎是TF 1.7中最近添加的预回迁到设备本问题提供了一些更详细的参考资料：

由于一些成本函数（如tf.nn.ctc_loss）在CPU上运行，因此不需要GPU内存上的标签，我们如何在小批量中只预取训练数据而不预取标签。

您需要创建两个数据集对象才能实现这一点。拥有两个数据集有很多用例，所以这并不异常

请注意，尽管预回迁不会预回迁到GPU，但这似乎是TF 1.7中最近添加的

预回迁到设备

本问题提供了一些更详细的参考资料：