mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

error when loadmodel a unexist file 

Signed-off-by: zhoufeng <zhoufeng54@huawei.com>
This commit is contained in:
zhoufeng 2021-02-23 12:52:30 +08:00
parent da6e6728b1
commit 5f6e15058d
2 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mindspore/ccsrc/cxx_api/graph/ascend/ascend_graph_impl.cc
View File

@ -24,6 +24,7 @@
#include "backend/session/session_factory.h"
#include "backend/session/executor_manager.h"
#include "runtime/device/kernel_runtime_manager.h"
#include "runtime/dev.h"
namespace mindspore {
API_FACTORY_REG(GraphCell::GraphImpl, Ascend910, AscendGraphImpl);

9
mindspore/ccsrc/cxx_api/serialization.cc
View File

@ -86,13 +86,10 @@ Graph Serialization::LoadModel(const void *model_data, size_t data_size, ModelTy
Graph Serialization::LoadModel(const std::string &file, ModelType model_type) {
if (model_type == kMindIR) {
FuncGraphPtr anf_graph = nullptr;
try {
anf_graph = LoadMindIR(file);
} catch (const std::exception &) {
MS_LOG(EXCEPTION) << "Load MindIR failed.";
FuncGraphPtr anf_graph = LoadMindIR(file);
if (anf_graph == nullptr) {
MS_LOG(EXCEPTION) << "Load model failed.";
}
return Graph(std::make_shared<Graph::GraphData>(anf_graph, kMindIR));
} else if (model_type == kOM) {
Buffer data = ReadFile(file);