Closed woodenwatcher closed 2 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)
不支持这种使用方式
想将paddle模型中conditional_block算子的“sub_block”属性里的子图导出成onnx,代码如下,其中第10行和11行之间应该将sub_block转为program类型的对象后再传递给run_convert,但我不知道如何去转换