Toybrick

标题: rknn-toolkit1.7.5 pytorch模型转换时 加载模型遇到问题 [打印本页]

作者: sfgzz    时间: 2024-9-20 15:59
标题: rknn-toolkit1.7.5 pytorch模型转换时 加载模型遇到问题
遇到的问题: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









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