3399pro 的 rknn toolkit load_onnx读取失败

[复制链接] · 发表于 2019-5-10 17:30:05

测试过weight.onnx是可以正常运行的，但使用rknn.load_onnx(model='./weight.onnx')后报出如下错误，这个错误不知道是由于模型不支持转换造成还是别的配置原因造成的？？我是用rknn-toolkit-0.9.9在3399pro板子上测试的，而且配置环境后在64位ubuntu上也是这个问题，希望能有人帮忙解答下

只看该作者 · 发表于 2019-5-10 17:50:06

看你取的名字貌似只有weights没有op，
确认下模型是完整操作+权重么？还是只有权重？

只看该作者 · 发表于 2019-5-10 18:08:27

我的onnx是pyTorch转过来的，pyTorch保存的时候是权重和模型结构分开保存，转化后的onnx好像也是权重。那我是不是要在保存pyTorch模型的时候就要采用权重和op同时保存的方式后面转化rknn才能成功？

只看该作者 · 发表于 2019-5-10 20:48:03

请问楼主weight.onnx怎么测试的？

3399pro 的 rknn toolkit load_onnx读取失败

本帖子中包含更多资源