Browse Source

fix ImageFolderDataset probability failure

tags/v1.0.0
jiangzhiwen 5 years ago
parent
commit
2a59596fe6
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/minddata/dataset/api/iterator.cc

+ 1
- 1
mindspore/ccsrc/minddata/dataset/api/iterator.cc View File

@@ -123,9 +123,9 @@ Status Iterator::BuildAndLaunchTree(std::shared_ptr<Dataset> ds) {

// Launch the execution tree.
RETURN_IF_NOT_OK(tree_->Prepare());
tree_->Launch();
iterator_ = std::make_unique<DatasetIterator>(tree_);
RETURN_UNEXPECTED_IF_NULL(iterator_);
RETURN_IF_NOT_OK(tree_->Launch());

return rc;
}


Loading…
Cancel
Save