Browse Source

fix tfrecord not enough data if shuffle is false

tags/v0.5.0-beta
yanghaitao 5 years ago
parent
commit
07877d1e65
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/dataset/engine/datasetops/source/tf_reader_op.cc

+ 1
- 1
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) {


Loading…
Cancel
Save