华为计算微信公众号
昇腾AI开发者公众号
华为计算微博
华为计算今日头条
yolov5s中添加了坐标注意力机制，导出的onnx模型中坐标注意力的最后一个操作是逐元素相乘算子Mul，前向函数内部也有一个算子Mul，转成om模型后可视化发现内部的Mul是Eltwise(NPU)，坐标注意力的最后一个Mul是BinaryMath(CPU)，导致我在3519dv500运行速度很慢，求助各位大佬
本帖最后由 匿名用户 于 2024/04/28 20:19:38 编辑
我要发帖子
点赞
收藏
回复
分享
举报
已解决 已解决
yolov5s中添加了坐标注意力机制，导出的onnx模型中坐标注意力的最后一个操作是逐元素相乘算子Mul，前向函数内部也有一个算子Mul，转成om模型后可视化发现内部的Mul是Eltwise(NPU)，坐标注意力的最后一个Mul是BinaryMath(CPU)，导致我在3519dv500运行速度很慢，求助各位大佬