Toybrick

标题: 3399pro 的 rknn toolkit load_onnx读取失败 [打印本页]

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

作者: jefferyzhang 时间: 2019-5-10 17:50
看你取的名字貌似只有weights没有op，
确认下模型是完整操作+权重么？还是只有权重？

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

作者: yaowei 时间: 2019-5-10 20:48
请问楼主weight.onnx怎么测试的？

欢迎光临 Toybrick (https://t.rock-chips.com/)