From 69b58e1ef51b0a09bd9a88dbe65e151e03bc08a6 Mon Sep 17 00:00:00 2001 From: changzherui Date: Mon, 19 Apr 2021 15:46:16 +0800 Subject: [PATCH] modify export onnx input --- mindspore/ccsrc/transform/express_ir/onnx_exporter.cc | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/mindspore/ccsrc/transform/express_ir/onnx_exporter.cc b/mindspore/ccsrc/transform/express_ir/onnx_exporter.cc index 382ad7be04..6b634be0cf 100644 --- a/mindspore/ccsrc/transform/express_ir/onnx_exporter.cc +++ b/mindspore/ccsrc/transform/express_ir/onnx_exporter.cc @@ -415,7 +415,7 @@ void OnnxExporter::InitModelInfo() { model_.set_producer_name("MindSpore"); model_.set_producer_version("1.0"); onnx::OperatorSetIdProto *opset_proto = model_.add_opset_import(); - opset_proto->set_version(9); + opset_proto->set_version(11); } void OnnxExporter::ExportFuncGraph(const FuncGraphPtr &func_graph, onnx::GraphProto *const graph_proto) { @@ -446,13 +446,14 @@ void OnnxExporter::ExportParameters(const FuncGraphPtr &func_graph, onnx::GraphP MS_LOG(EXCEPTION) << "Parameter '" << param->ToString() << "' could not cast to parameter."; } - onnx::ValueInfoProto *input_proto = graph_proto->add_input(); - input_proto->set_name(param_ptr->ToString()); - SetValueInfoType(param_ptr, input_proto); - + // set onnx input. if (!param_ptr->has_default()) { + onnx::ValueInfoProto *input_proto = graph_proto->add_input(); + input_proto->set_name(param_ptr->ToString()); + SetValueInfoType(param_ptr, input_proto); continue; } + // parameter with default value is an ONNX initializer onnx::TensorProto *initializer_proto = graph_proto->add_initializer(); initializer_proto->set_name(param_ptr->ToString());