×

签到

分享到微信

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

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

yolov5-6.1代码,在网络中加入CBAM注意力机制,在模型推理时报错 chrc2025-12-26 11:02:23 回复 6 查看 技术答疑 使用求助
yolov5-6.1代码,在网络中加入CBAM注意力机制,在模型推理时报错
分享到:

在yolov5-6.1代码的网络中添加CBAM模块,利用torch_gpu2mlu.py将yolov5代码进行转换,然后进行模型推理,但是在推理时报错,错误信息如下,请问如何解决?

RuntimeError: The internal kernel size for adaptive_max_pool2d_out_mlu should be smaller than 3582, while this kernel size is 6724image.png

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