当num epoch设置为None时，tensorflow训练无穷无尽

Question

我使用tf.estimator.inputs.numpy_input_fn生成输入函数，它是这样的：

train_input = tf.estimator.inputs.numpy_input_fn(x={'x': sst_train}, y=precip_train, shuffle=True, batch_size=100, num_epochs=None)

在训练过程中，我设置了一个for循环来控制它。这是代码：

for i in range(15):
est.train(input_fn=train_input, steps=20)

我认为它应该总共训练20 * 15次，但它最终会产生无限循环。这有什么问题？

Answer 1

“num_epochs：整数，迭代数据的纪元数。如果None将永远运行。”您可以在numpy_input_fn函数的api注释中看到这一点。