如何验证tensorflow 1.4的tf.data.Dataset是否缺乏？

Question

我正在进行一些并行运行的dataset.map(...)次呼叫。我的GPU没有100％运行，我怀疑预处理管道是责任。

如何在训练步骤之前验证管道是否缺乏，从而警告用户该情况？

我的最终数据集转换是dataset.prefetch(2)，它预取了2个完整批次。

例如，我希望这样做：

sess.run(iterator.is_ready)

或

sess.run(dataset.buffer_size)