!23866 fix the process not exit in the only mindData send data sence

Merge pull request !23866 from limingqi107/bug_fix
This commit is contained in:
i-robot 2021-09-22 22:45:08 +00:00 committed by Gitee
commit bbc7ab61b0
1 changed files with 1 additions and 0 deletions

View File

@ -1173,6 +1173,7 @@ bool InitExecDatasetVm(const std::string &queue_name, int64_t size, int64_t batc
}
#endif
MS_LOG(INFO) << "Start InitDataSet Entry";
mindspore::parse::python_adapter::set_python_env_flag(true);
ShapeVector int_input_indexes;
(void)std::transform(input_indexes.begin(), input_indexes.end(), std::back_inserter(int_input_indexes),
[](int64_t item) { return static_cast<int64_t>(item); });