如何使用tf.data数据集对同一批次执行多次训练操作

时间:2018-09-12 16:56:33

标签: python tensorflow

我想使用tf.data数据集和迭代器(对抗性自动编码器)实现here可用的代码。

我的问题是如何将同一批次用于多个训练操作?

我需要在同一批次上执行三种不同的训练操作,但是,如果我使用tf.data迭代器,则它不适用于同一批次,而不能用于连续批次。

1 个答案:

答案 0 :(得分:1)

假设data = db_iter.get_next(),其中db_iter是您从tf.data开始使用的迭代器。

我想您正在3不同的3语句中转换sess.run的操作。在这种情况下,他们将使用3个不同的批次,因为data将被评估3次。

事实是,如果三个training ops的每一个的输入是从上述data提供的,并且在每个sess.run中运行,它们将全部使用相同的批处理