×

签到

分享到微信

打开微信,使用扫一扫进入页面后,点击右上角菜单,

点击“发送给朋友”或“分享到朋友圈”完成分享

yolov5s模型进行寒武纪平台的移植报错 li7072024-01-19 16:26:04 回复 2 查看 技术答疑 使用求助
yolov5s模型进行寒武纪平台的移植报错
分享到:

我的卡是MLU270F4,麒麟V10系统,安装的pytorch1.3,最初的测试例程无报错,驱动无问题



本文对开源yolov5s模型进行寒武纪平台的移植

代码参考链接:https://github.com/CambriconECO/Pytorch_Yolov5_Inference/tree/eb36b68e5a4ee3fe23c45ec545bef68dced452bb


运行上面链接yolov5移植的程序detect.py,显示:

File "/home/ai-test2/Projects/Pytorch_Yolov5_Inference-master/quantize_online/yolo.py", line 56, in forward

    detect_out = torch.ops.torch_mlu.yolov5_detection_output(output[0], output[1], output[2],

  File "/home/ai-test2/miniconda3/lib/python3.7/site-packages/torch/_ops.py", line 61, in __getattr__

    op = torch._C._jit_get_operation(qualified_op_name)

RuntimeError: No such operator torch_mlu::yolov5_detection_output






版权所有 © 2024 寒武纪 Cambricon.com 备案/许可证号:京ICP备17003415号-1
关闭