×

签到

分享到微信

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

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

【MLU270】mlu pytorch不支持conv1d和batchnorm1d应该如何解决 已完结 Xinqian2022-07-12 19:54:19 回复 2 查看 使用求助
【MLU270】mlu pytorch不支持conv1d和batchnorm1d应该如何解决
分享到:

【寒武纪硬件产品型号】必填*:
MLU270


【使用操作系统】必填*:
ubuntu


【使用驱动版本】必填*:
v4.20.6


【使用AI框架】必填*:
Pytorch,


【操作步骤】选填:


【出错信息】必填*:

在做模型迁移的时候,网络里有很多conv1d和batchnorm1d的算子,但是发现官方文档里没有写支持这两个算子,直接用mlu的device上进行相关运算也会报错,请问针对这两个算子,应该如何解决呢?


【当前已做了哪些信息确认】选填:
例如交叉验证/CPU能运行/调试通过等


【参考配置文档链接】选填


【相关日志文档】选填
如有,可附件


【出错代码链接】选填:
github的或gitee的代码的链接,

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