如何使用存储桶为seq2seq模型设置tfrecords队列？

Question

我尝试将official tutorial关于seq2seq模型的输入从占位符转换为tfrecords但是失败主要是因为如果我将所有样本写入单个tfrecord文件，我就不能将具有相似长度的样本分组到同一个tfrecord文件中斗，我需要启动四个队列（桶的长度）才能实现它吗？欢迎任何建议和解决方案

Answer 1

新添加的tf.contrib.training.bucket_by_sequence_length()函数增加了对创建多个队列的支持，这些队列按分段序列长度进行分组，并将完整批次组合在一个队列中。