×
分享到微信

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

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

用示例代码和模型跑在MLU220_M.2上 memory报错 已解决 shimin.an2022-06-19 20:15:26 回复 2 查看 使用求助
用示例代码和模型跑在MLU220_M.2上 memory报错
分享到:

起因:在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时遇到的问题一样。

图片.png


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