标题: 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