Toybrick

标题: pytorch模型转换成功,但运行失败 [打印本页]

作者: qiu47    时间: 2021-2-8 17:36
标题: pytorch模型转换成功,但运行失败
--> 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    时间: 2021-2-8 18:13
具体模型问题欢迎到置顶帖rknn官方qq群讨论




欢迎光临 Toybrick (https://t.rock-chips.com/) Powered by Discuz! X3.3