当 num epoch 设置为 None 时，tensorflow 训练无休止答案

【问题标题】：tensorflow training endless when num epoch is set to None当 num epoch 设置为 None 时，tensorflow 训练无休止
【发布时间】：2018-05-08 05:25:59
【问题描述】：

我使用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 次，但它最终以无限循环结束。这里有什么问题？

【问题讨论】：

标签： python tensorflow machine-learning training-data

【解决方案1】：

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

【讨论】：

如果在estimator.train中设置了步数，这应该无关紧要。
是的，我想是的。当在 train 函数中设置步数时，它不应该永远运行。