PaddlePaddle / Paddle2ONNX

ONNX Model Exporter for PaddlePaddle
Apache License 2.0
717 stars 166 forks source link

如何将paddle模型中的sub_block导出成onnx #1217

Closed woodenwatcher closed 2 months ago

woodenwatcher commented 6 months ago

想将paddle模型中conditional_block算子的“sub_block”属性里的子图导出成onnx,代码如下,其中第10行和11行之间应该将sub_block转为program类型的对象后再传递给run_convert,但我不知道如何去转换


1. import paddle
2. import paddle2onnx
3.  
4. paddle.enable_static()
5. 
6. program = paddle.load("inference.pdmodel")
7. for block in porgram.blocks:
8.     if op.type == "conditional_block":
9.         sub_bock_id = op.attr("sub_block").id
10.        sub_block = block.program.block(sub_block_id)
11.        onnx_model = paddle2onnx.run_convert(sub_block)
Zheng-Bicheng commented 2 months ago

不支持这种使用方式