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

Merge pull request !1849 from yanghaitao/yht_fix_tf_noshuffle
pull/1849/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit 58643afc09

@ -481,7 +481,7 @@ Status TFReaderOp::FillIOBlockNoShuffle() {
int64_t start_offset = 0; int64_t start_offset = 0;
int64_t end_offset = 0; int64_t end_offset = 0;
bool finish = false; bool finish = false;
bool end_of_epoch = true; bool end_of_epoch = false;
while (!finish) { while (!finish) {
// Iterate over all the keys and add one key to each block. // Iterate over all the keys and add one key to each block.
for (auto it = filename_index_->begin(); it != filename_index_->end(); ++it) { for (auto it = filename_index_->begin(); it != filename_index_->end(); ++it) {

Loading…
Cancel
Save