当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注释中看到此numpy_input_fn 。