提交 07877d1e 编写于 作者: Y yanghaitao

fix tfrecord not enough data if shuffle is false

上级 e79df7c1
......@@ -481,7 +481,7 @@ Status TFReaderOp::FillIOBlockNoShuffle() {
int64_t start_offset = 0;
int64_t end_offset = 0;
bool finish = false;
bool end_of_epoch = true;
bool end_of_epoch = false;
while (!finish) {
// Iterate over all the keys and add one key to each block.
for (auto it = filename_index_->begin(); it != filename_index_->end(); ++it) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册