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

fix ImageFolderDataset probability failure

This commit is contained in:
jiangzhiwen 2020-09-12 17:05:06 +08:00
parent 02c6852699
commit 2a59596fe6
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/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;
}