版本是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 可以得到量化模型
请问我的操作步骤中问题在哪