ACL Runtime error code = [ 500002 ] ,input size < op_size /// is_input:0 blob size:1 from user != op_size:4 in model:1
收藏回复举报
ACL Runtime error code = [ 500002 ] ,input size < op_size /// is_input:0 blob size:1 from user != op_size:4 in model:1
发表于2023-12-06 22:43:30
0 查看

我是yolov5s的模型,输入维度是1*3*640*640,输出维度是1*25200*85,

使用下述转换的模型。

atc --input_shape="images:1,3,640,640" --input_format=NCHW --output="yolov5s" --soc_version=Ascend310B4 --framework=5 --model="yolov5s.onnx" --output_type=FP32

推理时会报错,input size是1228800和op_size的4915232没对齐

cke_15884.png

然后我就手动分配input buffer的内存大小为4915232,然后就会报is_input:0 blob size:1 user !=op_size

cke_170.png

目前试用了各类atc转换代码,都是这两个错误,不知道怎么解决。

onnx模型和 soc_version=Ascend310B4 条件下转出的om文件在下述链接。

链接:https://pan.baidu.com/s/13WQHuZMDjFdCzuv2zL9PNw?pwd=udo2  

提取码:udo2  

--来自百度网盘超级会员V5的分享

本帖最后由 匿名用户2023/12/07 09:03:34 编辑

我要发帖子