Toybrick

rknn-toolkit1.7.5 pytorch模型转换时 加载模型遇到问题

sfgzz

新手上路

积分
15
楼主
发表于 2024-9-20 15:59:09    查看: 746|回复: 0 | [复制链接]    打印 | 只看该作者
遇到的问题:1. load pytorch报错 E ValueError: need more than 1 value to unpack
具体如下所示,网络模型的解析完成后报错。请问出现这个报错的可能原因是什么呢
I Parse aten::add at 796
D Output shape of layer add_at_796_272_202: [[1, 1, 256, 256]]
I Parse intermediate nodes finished.
I Parse pytorch model finished!
E Catch exception when loading pytorch model: /home/rknn/model/fusion/fusion.pt!
E Traceback (most recent call last):
E   File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 478, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build_layer
E   File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 478, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build_layer
E   File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 478, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build_layer
E   [Previous line repeated 1 more time]
E   File "rknn/base/RKNNlib/RKNNnetbuilder.py", line 476, in rknn.base.RKNNlib.RKNNnetbuilder.RKNNNetBuilder.build_layer
E ValueError: need more than 1 value to unpack
E Please feedback the detailed log file <log_feedback_to_the_rknn_toolkit_dev_team.log> to the RKNN Toolkit development team.
E You can also check github issues: https://github.com/rockchip-linux/rknn-toolkit/issues




回复

使用道具 举报

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

本版积分规则

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


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