pytorch 的 nn.Upsample 和 nn.MaxPool2d 是否支持转成 rknn?

我也遇到这个问题了，pytorch->onnx>rknn。torch.nn.Upsample报错。请问哪个现在哪个版本支持，如果不支持，我们自己改如何改进。

奥古师弟 发表于 2020-6-15 17:45
我也遇到这个问题了，pytorch->onnx>rknn。torch.nn.Upsample报错。请问哪个现在哪个版本支持，如果不支持 ...

之前没看到最新回复。

jefferyzhang 发表于 2020-5-9 08:22
NPU Team 答复：
请提供下保存下来的pt模型和onnx模型，pytorch版本建议1.2.0，onnx建议1.4.1

安装要求修改了unsample，但是还是报了这个错误。  pytorch->onnx->rknn.load_onnx()

E Traceback (most recent call last):
E   File "rknn/base/RKNNlib/converter/convert_onnx.py", line 1071, in rknn.base.RKNNlib.converter.convert_onnx.convert_onnx.match_paragraph_and_param
E   File "rknn/base/RKNNlib/converter/convert_onnx.py", line 980, in rknn.base.RKNNlib.converter.convert_onnx.convert_onnx._onnx_push_ready_tensor
E TypeError: 'NoneType' object is not iterable
E During handling of the above exception, another exception occurred:
E Traceback (most recent call last):
E   File "rknn/api/rknn_base.py", line 513, in rknn.api.rknn_base.RKNNBase.load_onnx
E   File "rknn/base/RKNNlib/converter/convert_onnx.py", line 1077, in rknn.base.RKNNlib.converter.convert_onnx.convert_onnx.match_paragraph_and_param
E   File "rknn/api/rknn_log.py", line 312, in rknn.api.rknn_log.RKNNLog.e
E ValueError: Try match Upsample_270ut0 failed, catch exception!