diff --git a/mindspore/ccsrc/pipeline/pipeline.cc b/mindspore/ccsrc/pipeline/pipeline.cc index 7b3cf06b5eb..3c3478e89d2 100644 --- a/mindspore/ccsrc/pipeline/pipeline.cc +++ b/mindspore/ccsrc/pipeline/pipeline.cc @@ -1110,10 +1110,11 @@ bool ExecutorPy::AddDFGraph(const py::dict& init_params, const std::string& phas (void)DfGraphManager::GetInstance().AddGraph(phase, convertor.GetComputeGraph()); } (void)DfGraphManager::GetInstance().AddGraph(init_graph, convertor.GetInitGraph()); - (void)DfGraphManager::GetInstance().AddGraph(checkpoint_name, convertor.GetSaveCheckpointGraph()); (void)DfGraphManager::GetInstance().AddGraph(BROADCAST_GRAPH_NAME, convertor.GetBroadcastGraph()); - - DfGraphManager::GetInstance().SetAnfGraph(checkpoint_name, anf_graph); + Status ret = DfGraphManager::GetInstance().AddGraph(checkpoint_name, convertor.GetSaveCheckpointGraph()); + if (ret == Status::SUCCESS) { + DfGraphManager::GetInstance().SetAnfGraph(checkpoint_name, anf_graph); + } return true; } diff --git a/mindspore/ccsrc/transform/convert.cc b/mindspore/ccsrc/transform/convert.cc index c975b18812e..2b50d2328a2 100755 --- a/mindspore/ccsrc/transform/convert.cc +++ b/mindspore/ccsrc/transform/convert.cc @@ -1104,12 +1104,12 @@ void DfGraphConvertor::UpdateDataOpDesc(const AnfNodePtr &it, const OperatorPtr auto normal_shape_ptr = dyn_cast(node->Shape()); vector shape; if (normal_shape_ptr == nullptr) { - MS_LOG(ERROR) << "Update data op descriptor failed! Invalid shape."; + MS_LOG(INFO) << "Invalid shape to update data op descriptor."; return; } shape = normal_shape_ptr->shape(); if (node->Type() == nullptr) { - MS_LOG(ERROR) << "Update data op descriptor failed! Invalid type."; + MS_LOG(INFO) << "Invalid type to update data op descriptor."; return; } TypeId me_type = node->Type()->type_id(); diff --git a/mindspore/ccsrc/transform/df_graph_manager.cc b/mindspore/ccsrc/transform/df_graph_manager.cc index 3339f431451..bfe4d9f5d23 100644 --- a/mindspore/ccsrc/transform/df_graph_manager.cc +++ b/mindspore/ccsrc/transform/df_graph_manager.cc @@ -112,7 +112,7 @@ DfGraphWrapperPtr DfGraphManager::GetGraphByName(const std::string& name) { auto it = graphs_.find(name); if (it == graphs_.end()) { - MS_LOG(ERROR) << "Can't found graph name: " << name; + MS_LOG(INFO) << "Can't found graph name: " << name; return nullptr; } MS_LOG(INFO) << "Return graph: " << name;