neural-network - TensorFlow 翻译培训——什么时候停止？

Question

我正在使用来自 Google 的 Tensorflow 的示例。我能够很好地运行它，但训练似乎并没有停止，因为它有一个while True:循环。参考： translate.py

考进train()：

while True:
  # Choose a bucket according to data distribution. We pick a random number
  # in [0, 1] and use the corresponding interval in train_buckets_scale.
  random_number_01 = np.random.random_sample()
  bucket_id = min([i for i in xrange(len(train_buckets_scale))
                   if train_buckets_scale[i] > random_number_01])
....

Answer 1

这是真的，在某一时刻，您可以中止训练过程。它的时刻取决于您选择的模型参数。根据 tensorflow文档，它需要大约 340K 步，选择的批量大小为 64。如果选择了创建检查点的选项，您可以使用任何检查点来检查您的模型与解码器的性能。