|
本帖最后由 jefferyzhang 于 2020-5-9 11:28 编辑
NPU Team 答复:
请提供下保存下来的pt模型和onnx模型,pytorch版本建议1.2.0,onnx建议1.4.1
updated:
这两个op现在rknn toolkit都是支持的,但是upsample支持的不是很完整,目前只支持resize参数,不支持scale_factor参数,我们会在下一个版本完善对这个OP的支持。
upsample可以参考附件upsample_demo.py的写法来定义网络。maxpool转onnx后也是可以的,写法参考maxpool_demo.py。请确保pytorch版本是1.2.0,onnx版本是1.4.1.
附:以上脚本在RKNN Toolkit 1.3.2上测试。
(250082)
(#250082) |
本帖子中包含更多资源
您需要 登录 才可以下载或查看,没有帐号?立即注册
x
|