标题: W Input channel 1 less then 3, please update NPU Driver to v1.3.0 or later. [打印本页] 作者: 午后阳光 时间: 2020-10-11 11:07 标题: W Input channel 1 less then 3, please update NPU Driver to v1.3.0 or later. 本帖最后由 午后阳光 于 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!