From 8d907a3c7153700e59e0289924ec27d9e0151582 Mon Sep 17 00:00:00 2001 From: changzherui Date: Tue, 16 Mar 2021 12:58:47 +0800 Subject: [PATCH] modify export mindir --- mindspore/core/load_mindir/load_model.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/core/load_mindir/load_model.cc b/mindspore/core/load_mindir/load_model.cc index 2d5ca68e6f9..02a8f7fe632 100644 --- a/mindspore/core/load_mindir/load_model.cc +++ b/mindspore/core/load_mindir/load_model.cc @@ -132,7 +132,7 @@ std::shared_ptr LoadMindIR(const std::string &file_name, bool is_lite } // Load parameter into graph - if (endsWith(abs_path_buff, "_graph.mindir")) { + if (endsWith(abs_path_buff, "_graph.mindir") && origin_model.graph().parameter_size() == 0) { int path_len = strlen(abs_path_buff) - strlen("graph.mindir"); memcpy(abs_path, abs_path_buff, path_len); abs_path[path_len] = '\0';