Toybrick

rknn模型,python端和c端模型要求的输入数据类型不一致

Lee

注册会员

积分
187
楼主
发表于 2020-12-1 10:01:38    查看: 6243|回复: 0 | [复制链接]    打印 | 只看该作者
如题,基于最新的rknn-toolkit-1.4工具,pytorch模型转rknn模型,未开量化,rk3399pro,api:1.4.0,drv:1.5.0;通过pc连接3399,模型跑在3399上,输入是float32,即rknn.inference接口的data_type选其他类型会出错。但在3399上跑c代码,发现模型要求的输入变成了float16,这是为什么呢?
回复

使用道具 举报

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

本版积分规则

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


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