rknn.load_pytorch 模型加载接口报错

请问下pyTorch的版本号是多少

ivanzheng 发表于 2020-4-20 11:20
pytorch 版本：1.2.0

解决方法：
1. 我会把这个op报给相关部门支持，要等未来版本出来大家才能用到。
2. 自己修改模型绕过这个OP
3. 用其他的框架试试，例如pytorch转onnx再转rknn
4. 尝试自己支持，我们rknn支持自定op，可以自己根据op的逻辑写一个op出来。

ivanzheng 发表于 2020-4-20 11:41
好的，感谢， 下一个发布版本会支持嘛？ 下个版本预计什么时候发布哈?
另外， 如果是tensorflow 实现的这 ...

这个我也不确定，我不是NPU部门的，我也只是一个rknn的开发者之一。只能说试试。

NPU部门回复：

1.请用1.3.3_beta2版本：
链接: https://pan.baidu.com/s/12ayycMiZGvJnpcD13DmH9g 密码: fe2w
2.rknn.config里加上optimization_level=2