Распечатать ONNXRUNTIME::Graph без модели

Question

Я работаю над новым оптимизатором, который наследуется от GraphTransformer в C++

Как я могу просмотреть внесенные мной изменения сразу после завершения 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");