Toybrick

W Input channel 1 less then 3, please update NPU Driver to v1.3.0 or later.

午后阳光

注册会员

积分
111
楼主
发表于 2020-10-11 11:07:34    查看: 6732|回复: 0 | [复制链接]    打印 | 只看该作者
本帖最后由 午后阳光 于 2020-11-2 12:17 编辑

有人知道这个报错原因吗?万分感谢,用torch.jit.trace的得到的模型用Netron打开没有问题,附图如图片所示,转换rknn模型环境:pytorch1.2.0,rknn-toolkit-v1.4.0,在rknn.load_pytorch时报错
WARNING: Token 'COMMENT' defined, but not used
WARNING: There is 1 unused token
W Input channel 1 less then 3, please update NPU Driver to v1.3.0 or later.
W Input channel 1 less then 3, please update NPU Driver to v1.3.0 or later.
W Input channel 1 less then 3, please update NPU Driver to v1.3.0 or later.
E Catch exception when loading pytorch model: /home/ganhaiyang/Alg_Proj/Recog_Proj/TSM-Detection/weights/tsmsfNetron.pt!
E Traceback (most recent call last):
E   File "rknn/api/rknn_base.py", line 657, in rknn.api.rknn_base.RKNNBase.load_pytorch
E   File "rknn/base/RKNNlib/app/importer/import_pytorch.py", line 97, in rknn.base.RKNNlib.app.importer.import_pytorch.ImportPytorch.run
E   File "rknn/base/RKNNlib/converter/convert_pytorch.py", line 755, in rknn.base.RKNNlib.converter.convert_pytorch.convert_pytorch.__init__
E   File "rknn/base/RKNNlib/converter/convert_pytorch.py", line 839, in rknn.base.RKNNlib.converter.convert_pytorch.convert_pytorch.model_simplify
E   File "rknn/base/RKNNlib/converter/convert_pytorch.py", line 124, in rknn.base.RKNNlib.converter.convert_pytorch.torch_inference_engine.shape_pick
E   File "rknn/base/RKNNlib/converter/convert_pytorch.py", line 159, in rknn.base.RKNNlib.converter.convert_pytorch.torch_inference_engine.__ir_shape_inference
E KeyError: 'aten::to'
Load pytorch model failed!

回复

使用道具 举报

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

本版积分规则

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


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