海思NNIE reshape only support C/H/W,not N!

海思NNIE reshape only support C/H/W,not N!_第1张图片


简单的一个google net,转成caffe。

 这是proto

海思NNIE reshape only support C/H/W,not N!_第2张图片

这是ruyi 量化时报的错


caffe 转wk 遇到了这个问题:reshape only support C/H/W,not N!

 我刚刚用ruyi 的python 环境和那个caffe模型 用个python 脚本推理了一下 ,没报错呀,说明python 环境和模型都是没问题的,应该是ruyi量化的时候出了问题

解决办法把dim : 1 改成 0

最关键的是,改了proto 之后要在重新browse 那个文件,让ruyi重新根据proto生成东西。

我尝试过删掉dim 1 也不行

你可能感兴趣的:(海思AI,caffe,人工智能,深度学习)