p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix tfrecord not enough data if shuffle is false

This commit is contained in:
yanghaitao 2020-06-04 17:32:45 +08:00
parent e79df7c169
commit 07877d1e65
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/dataset/engine/datasetops/source/tf_reader_op.cc
View File

@ -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) {