diff --git a/mindspore/lite/examples/transfer_learning/src/dataset.cc b/mindspore/lite/examples/transfer_learning/src/dataset.cc index 1a52da7f93..7a0669bb12 100644 --- a/mindspore/lite/examples/transfer_learning/src/dataset.cc +++ b/mindspore/lite/examples/transfer_learning/src/dataset.cc @@ -202,8 +202,10 @@ std::vector DataSet::ReadDir(const std::string dpath) { vec.push_back(ft); } } + closedir(dp); + } else { + std::cerr << "open directory: " << dirname << " failed." << std::endl; } - closedir(dp); } return vec; }