算能科技
直播中

王秀珍

7年用户 1430经验值
私信 关注
[问答]

yolov5s_v6.1_3output.torchscript.pt模型auto_cali一键量化工具执行报错的原因?

版本是230501
torch的版本是v1.8.0+cpu
下载yolov5官方的yolov5s.pt
执行 python3 export.py --weights ./yolov5s.pt --include torchscript 得到 yolov5s_v6.1_3output.torchscript.pt  文件
执行 ./gen_int8bmodel_nntc.sh BM1684 报错,参见下图附件
ps: 直接用算法脚本下载yolov5s_v6.1_3output.torchscript.pt然后执行 ./gen_int8bmodel_nntc.sh BM1684 可以得到量化模型
请问我的操作步骤中问题在哪

更多回帖

发帖
×
20
完善资料,
赚取积分