!1849 fix tfrecord bug:less data if shuffle is false

Merge pull request !1849 from yanghaitao/yht_fix_tf_noshuffle
5 years ago · 58643afc09
parent 05c1b36c51 07877d1e65
commit 58643afc09
1 changed files with 1 additions and 1 deletions
--- a/mindspore/ccsrc/dataset/engine/datasetops/source/tf_reader_op.cc
+++ b/mindspore/ccsrc/dataset/engine/datasetops/source/tf_reader_op.cc
@ -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) {