tf.train.string_input_producer（）中的“capacity”参数是什么

Question

函数capacity的{​​{1}}参数对我来说相当含糊。

在这里设置tf.train.string_input_producer(string_tensor, num_epochs=None, shuffle=True, seed=None, capacity=32, shared_name=None, name=None)参数是什么意思，它与参数capacity的长度有什么关系。 一个例子将是美好的。 非常感谢。

Answer 1

容量是队列的大小，因此在您的示例中，队列运行器默认情况下最多可以将32个字符串排入队列

Answer 2

string_input_producer的容量是队列在任何给定时间可以容纳的最大元素数。您应该将数字设置得足够高，这样您的模型就不会因数据不足而挨饿。但如果你将它设置得太高，队列将消耗太多内存。

最佳数字是特定于模型的，您可以通过反复试验找到它。从一个相当小的数字开始，检查您的队列空的频率。增加缓冲区，直到看不到空队列。