From 2a59596fe61124e788521c2638c3de6faf75b2f9 Mon Sep 17 00:00:00 2001
From: jiangzhiwen <jiangzhiwen8@huawei.com>
Date: Sat, 12 Sep 2020 17:05:06 +0800
Subject: [PATCH] fix ImageFolderDataset probability failure

---
 mindspore/ccsrc/minddata/dataset/api/iterator.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mindspore/ccsrc/minddata/dataset/api/iterator.cc b/mindspore/ccsrc/minddata/dataset/api/iterator.cc
index 1559ca24c5..cfdfcb8bc8 100644
--- a/mindspore/ccsrc/minddata/dataset/api/iterator.cc
+++ b/mindspore/ccsrc/minddata/dataset/api/iterator.cc
@@ -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;
 }