diff --git a/mindspore/core/load_mindir/load_model.cc b/mindspore/core/load_mindir/load_model.cc index 1b457e5c08..2d5ca68e6f 100644 --- a/mindspore/core/load_mindir/load_model.cc +++ b/mindspore/core/load_mindir/load_model.cc @@ -127,7 +127,7 @@ std::shared_ptr LoadMindIR(const std::string &file_name, bool is_lite mind_ir::ModelProto origin_model; if (!input_graph || !origin_model.ParseFromIstream(&input_graph)) { - MS_LOG(ERROR) << "Load MindIR file failed."; + MS_LOG(ERROR) << "Load MindIR file failed, please check the correctness of the file."; return nullptr; } @@ -143,7 +143,7 @@ std::shared_ptr LoadMindIR(const std::string &file_name, bool is_lite string path = abs_path; get_all_files(path, &files); } else { - MS_LOG(ERROR) << "MindIR graph has not variable path, load failed"; + MS_LOG(ERROR) << "Load graph's variable folder failed, please check the correctness of variable folder."; return nullptr; } @@ -153,7 +153,7 @@ std::shared_ptr LoadMindIR(const std::string &file_name, bool is_lite std::fstream input_param(files[file_index], std::ios::in | std::ios::binary); mind_ir::GraphProto param_graph; if (!input_param || !param_graph.ParseFromIstream(&input_param)) { - MS_LOG(ERROR) << "Load param proto file failed."; + MS_LOG(ERROR) << "Load variable file failed, please check the correctness of mindir's variable file."; return nullptr; } diff --git a/mindspore/train/serialization.py b/mindspore/train/serialization.py index c34bbe332c..3f25e60e12 100644 --- a/mindspore/train/serialization.py +++ b/mindspore/train/serialization.py @@ -721,7 +721,7 @@ def _save_mindir(net, file_name, *inputs): if data_size > TOTAL_SAVE: data_file_name = data_path + "/" + "data_" + str(index) with open(data_file_name, "ab") as f: - os.chmod(file_name, stat.S_IRUSR | stat.S_IWUSR) + os.chmod(data_file_name, stat.S_IRUSR | stat.S_IWUSR) f.write(graphproto.SerializeToString()) index += 1 data_size = 0 @@ -730,7 +730,7 @@ def _save_mindir(net, file_name, *inputs): if graphproto.parameter: data_file_name = data_path + "/" + "data_" + str(index) with open(data_file_name, "ab") as f: - os.chmod(file_name, stat.S_IRUSR | stat.S_IWUSR) + os.chmod(data_file_name, stat.S_IRUSR | stat.S_IWUSR) f.write(graphproto.SerializeToString()) # save graph