Toybrick

请问目前可以Pytorch到rknn的转换吗

jefferyzhang

版主

积分
12952
楼主
发表于 2020-11-17 09:03:04 | 显示全部楼层
sample里就有,目前的rknntoolkit版本(1.4.0)pytorch支持到1.2.0.
请先仔细看下toolkit的文档
回复

使用道具 举报

jefferyzhang

版主

积分
12952
沙发
发表于 2020-11-20 11:35:02 | 显示全部楼层
changchenliang 发表于 2020-11-20 11:06
你好,不好意思我想再请教一下,您和toolkit文档里都写了对应的pytorch安装是1.2.0, 请问这意思是包括1. ...

只支持1.2.0.
其他版本高了低了都可能会有奇怪问题,当然也可能运气好能正常使用。
回复

使用道具 举报

jefferyzhang

版主

积分
12952
板凳
发表于 2020-11-20 14:11:54 | 显示全部楼层
changchenliang 发表于 2020-11-20 11:47
谢谢!再麻烦请教下:
我在论坛下载了rknn_toolkit 1.4.0压缩包,里面的package文件夹下的7个whl轮子安装 ...

以发布包支持的平台和版本为准
回复

使用道具 举报

jefferyzhang

版主

积分
12952
地板
发表于 2020-11-25 16:16:51 | 显示全部楼层
changchenliang 发表于 2020-11-25 16:14
您好,请问一下pytorch模型的训练,保存及转rknn是不是都要求是1.2.0版本?
如果我再PC上训练用的是高版 ...

目前rknn-toolkit(1.4.0)仅支持到pytorch 1.2.0,
其余版本高了低了均不可以转换。

只要你pytorch生成的模型在torch 1.2.0能读出来并用1.2.0版本pytorch能正确推理就可以。
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表