Toybrick

标题: 安装更新了RKNN1.4运行官方例子pytorch的resnet18时出错 [打印本页]

作者: xiangqian    时间: 2020-9-30 08:11
标题: 安装更新了RKNN1.4运行官方例子pytorch的resnet18时出错
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!


请问怎么解决,谢谢!


作者: geoergecyr    时间: 2020-11-16 21:50
我在转换模型的过程中也遇到了这个问题,但是我换成rknn 1.3.2之后就能没有这个问题了。
作者: segust    时间: 2020-12-16 19:53
torchvision安装0.4.0版本
作者: 小宁哥    时间: 2020-12-18 13:34
segust 发表于 2020-12-16 19:53
torchvision安装0.4.0版本

我的torchvision最新 只有0.2.2post3,是要自己下载轮子装吗
作者: segust    时间: 2020-12-21 15:49
恩,是的




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