Toybrick

楼主: wujialiang

pytorch模型转换问题

jefferyzhang

版主

积分
12937
31#
发表于 2020-4-6 08:26:33 | 只看该作者
rlgl 发表于 2020-4-5 15:33
您好,beta7版本nn.Transpose2d输入输出维度正常。但另一个问题:在使用相同输入情况下,pytorch模型输出 ...

不量化差异大么?
如果不量化差异也大,就要dump出每一层看下哪层有问题。
如果量化差异大,那就用混合量化,把差异大的层改成不量化
回复

使用道具 举报

rlgl

注册会员

积分
101
32#
发表于 2020-4-9 19:25:22 | 只看该作者
问个与本问题无关的问题,如何将模型运行在与pc相连的rk3399pro上,我看到Rockchip_User_Guide_RKNN_Toolkit_V1.3.0_CN.pdf文档的3.2.2节写道:“确保开发板的 USB OTG 连接到 PC,并且正确识别到设备,即在 PC 上调用 RKNN-Toolkit的 list_devices 接口可查询到相应的设备”。
请问这个USB OTG 连接到 PC是什么意思,是通过开发板上的那个接口与pc连接
回复

使用道具 举报

top123

新手上路

积分
20
33#
发表于 2020-8-21 11:32:54 | 只看该作者
wujialiang 发表于 2020-2-14 17:17
什么时候rknnAPI可以在rk1808上支持c++直接使用pytorch的模型?目前我使用的detectron2训练的mask rcnn  ...

请问你转mask rcnn有转成功吗  我用pytorch自带的mask rcnn都没转成功
回复

使用道具 举报

top123

新手上路

积分
20
34#
发表于 2020-8-21 11:46:33 | 只看该作者
jefferyzhang 发表于 2020-2-16 19:42
rknn-toolkit存在的目的就是倒入其他格式模型,转换成rknn。
rknn-toolkit pc版和arm版功能几乎是一样的 ...

有在pytorch1.2中将torchvision.models.detection.maskrcnn_resnet50_fpn的maskrcnn模型转换示例吗
回复

使用道具 举报

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

本版积分规则

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


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