Toybrick

求助,pytorch模型转换成功,但推理异常

Lee

注册会员

积分
166
发表于 3 天前    查看: 32|回复: 2 | [复制链接]    打印 | 显示全部楼层
硬件版本号:api:1.4.0,drv:1.3.1,最新的rknn-toolkit1.4.0.

输入为单通道图像,且是归一化的浮点数(减mean,除以std),config参数为空,未打开量化,模型转换成功,但inference异常,和直接torch.jit.load加载运行的结果相差很多,像是溢出值,大量的65504之类。

debug一下,将输入rknn模型的数据在进入网络第一层之前,直接return,打印出来的值,依然是大量类似65504的值。

是版本的问题嘛?求助,谢谢!

回复

使用道具 举报

jefferyzhang

超级版主

积分
6467
发表于 前天 12:09 | 显示全部楼层
pc仿真试下,仿真如果也是错的就是rknn问题,仿真对的开发版跑错的就是驱动版本问题
回复

使用道具 举报

Lee

注册会员

积分
166
 楼主| 发表于 昨天 18:00 | 显示全部楼层
jefferyzhang 发表于 2020-11-22 12:09
pc仿真试下,仿真如果也是错的就是rknn问题,仿真对的开发版跑错的就是驱动版本问题 ...

已解决,确实是驱动问题,更新到最新的drv1.5就解决啦,多谢!
回复

使用道具 举报

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

本版积分规则

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


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