Toybrick

pytorch模型转换成功,但运行失败

qiu47

注册会员

积分
124
楼主
发表于 2021-2-8 17:36:38    查看: 11075|回复: 1 | [复制链接]    打印 | 只看该作者
--> Init runtime environment
E [op_check:323]Concat input dims size(39 vs 40)
E [setup_node:449]Check node[58] CONCAT fail
Create RKNN model fail, error=-6
E Catch exception when init runtime!
E Traceback (most recent call last):
E   File "rknn/api/rknn_base.py", line 844, in rknn.api.rknn_base.RKNNBase.init_runtime
E   File "rknn/api/rknn_runtime.py", line 340, in rknn.api.rknn_runtime.RKNNRuntime.build_graph
E Exception: RKNN init failed. error code: RKNN_ERR_MODEL_INVALID
Init runtime environment failed

原模型在pytorch框架下能够正常执行,但转换后执行中featuremap尺寸发生了变话(应该都是40)

回复

使用道具 举报

jefferyzhang

版主

积分
13580
沙发
发表于 2021-2-8 18:13:18 | 只看该作者
具体模型问题欢迎到置顶帖rknn官方qq群讨论
回复

使用道具 举报

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

本版积分规则

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


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