c++ - 在没有模型的情况下打印 ONNXRUNTIME::Graph

Question

我正在开发从C++ 中的GraphTransformer继承的新优化器

如何查看我所做的更改，就在ApplyImpl结束后，我尝试了 graph.ToGraphProto()但我无法保存它而且我需要创建一个新模型才能保存它并在 Netron 或其他东西中查看它

Answer 1

发现需要制作新模型和版本需要获取SchemaRegistry：

ModelProto model_proto1;
model_proto1.set_ir_version(IR_VERSION);
OperatorSetIdProto* op_set_id = model_proto1.add_opset_import();
op_set_id->set_domain("");
op_set_id->set_version(12);
model_proto1.set_allocated_graph(new GraphProto(graph.ToGraphProto()));
IOnnxRuntimeOpSchemaRegistryList local_registries{graph.GetSchemaRegistry()};


Model model(model_proto1, &local_registries, logger);
onnxruntime::Model::Save(model, "/tmp/new_g.onnx");