打开微信,使用扫一扫进入页面后,点击右上角菜单,
点击“发送给朋友”或“分享到朋友圈”完成分享
起因:在MLU220_M.2上尝试用yolov5s-6.0推理,无法生成离线模型,并且执行推理会报memory的错误,之后没有定位到问题原因。
于是尝试用官网提供的demo跑了一下。
git hub上下载的示例代码:https://developer.cambricon.com/index/curriculum/expdetails/id/10/classid/8.html
测试环境:MLU220_M.2
示例代码,只修改了detect.py, 在 Line129 添加:ct.set_core_version('MLU220')
运行:python detect.py --jit True --save True
结果:无法生成离线模型
并且:在执行推理:detect_out = quantized_net(img.to(ct.mlu_device()))时会报错,与我自己用最新yolov5s时遇到的问题一样。
热门帖子
精华帖子