×

签到

分享到微信

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

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

【MagicMind】模型推理失败,output shape 错误 luke_xu2023-12-26 15:40:22 回复 查看 技术答疑 使用求助
【MagicMind】模型推理失败,output shape 错误
分享到:

【寒武纪硬件产品型号】必填*:
必填项,例如:MLU370


【MagicMind版本号】必填*:

1.5.0

【出错信息】必填*:

image.png

image.png



【操作步骤】选填:

原模型为tensorflow1.15-gpu 训练DIEN模型得到的 checkpoint文件,后将checkpoint固化转为pb文件,magicmind加载解析此pb文件。

由“

output_dims = model.get_output_dimensions()


输出output_dims为Dims([]),并有如上错误信息。

这种现象是相关的算子导致的shape infer error么,有什么其它路径可以规避此error


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


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

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