安装更新了RKNN1.4运行官方例子pytorch的resnet18时出错

[复制链接] · 发表于 2020-9-30 08:11:27

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 File "rknn/base/RKNNlib/converter/convert_pytorch.py", line 398, in rknn.base.RKNNlib.converter.convert_pytorch.torch_inference_engine.addmm_shape
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 File "rknn/base/RKNNlib/converter/convert_pytorch.py", line 281, in rknn.base.RKNNlib.converter.convert_pytorch.torch_inference_engine.view_shape
E ZeroDivisionError: integer division or modulo by zero
Load pytorch model failed!

请问怎么解决，谢谢！

只看该作者 · 发表于 2020-11-16 21:50:47

我在转换模型的过程中也遇到了这个问题，但是我换成rknn 1.3.2之后就能没有这个问题了。

只看该作者 · 发表于 2020-12-16 19:53:28

torchvision安装0.4.0版本

只看该作者 · 发表于 2020-12-18 13:34:30

segust 发表于 2020-12-16 19:53
torchvision安装0.4.0版本

我的torchvision最新只有0.2.2post3，是要自己下载轮子装吗

只看该作者 · 发表于 2020-12-21 15:49:28

恩，是的