rknn.load_pytorch 模型加载接口报错

Hi 你好，
我在将pytorch 模型转换成 rknn模型时，出现以下错误， 看错误信息好像是PixelShuffle 算子不支持， 如果想把这个模型（超分辨率）跑在 rk3399pro 上面，需要如何处理，请指教，多谢

rknn 版本： 1.3

模型repo：
https://github.com/thstkdgus35/EDSR-PyTorch

错误信息：
E Catch exception when loading pytorch model: ./edsr_baseline_x2_pytorch.pt!
E Traceback (most recent call last):
E   File "rknn/api/rknn_base.py", line 567, in rknn.api.rknn_base.RKNNBase.load_pytorch
E   File "rknn/base/RKNNlib/app/importer/import_pytorch.py", line 95, in rknn.base.RKNNlib.app.importer.import_pytorch.ImportPytorch.run
E   File "rknn/base/RKNNlib/converter/convert_pytorch.py", line 517, in rknn.base.RKNNlib.converter.convert_pytorch.convert_pytorch.__init__
E   File "rknn/base/RKNNlib/converter/convert_pytorch.py", line 601, in rknn.base.RKNNlib.converter.convert_pytorch.convert_pytorch.model_simplify
E   File "rknn/base/RKNNlib/converter/convert_pytorch.py", line 104, in rknn.base.RKNNlib.converter.convert_pytorch.torch_inference_engine.shape_pick
E   File "rknn/base/RKNNlib/converter/convert_pytorch.py", line 139, in rknn.base.RKNNlib.converter.convert_pytorch.torch_inference_engine.__ir_shape_inference
E   File "rknn/base/RKNNlib/converter/convert_pytorch.py", line 247, in rknn.base.RKNNlib.converter.convert_pytorch.torch_inference_engine.convolution_shape
E   File "rknn/base/RKNNlib/converter/convert_pytorch.py", line 104, in rknn.base.RKNNlib.converter.convert_pytorch.torch_inference_engine.shape_pick
E   File "rknn/base/RKNNlib/converter/convert_pytorch.py", line 139, in rknn.base.RKNNlib.converter.convert_pytorch.torch_inference_engine.__ir_shape_inference
E   File "rknn/base/RKNNlib/converter/convert_pytorch.py", line 247, in rknn.base.RKNNlib.converter.convert_pytorch.torch_inference_engine.convolution_shape
E   File "rknn/base/RKNNlib/converter/convert_pytorch.py", line 104, in rknn.base.RKNNlib.converter.convert_pytorch.torch_inference_engine.shape_pick
E   File "rknn/base/RKNNlib/converter/convert_pytorch.py", line 139, in rknn.base.RKNNlib.converter.convert_pytorch.torch_inference_engine.__ir_shape_inference
E KeyError: 'aten::pixel_shuffle'
Load pytorch model failed!

jefferyzhang 发表于 2020-4-20 11:05
请问下pyTorch的版本号是多少

pytorch 版本：1.2.0

jefferyzhang 发表于 2020-4-20 11:29
解决方法：
1. 我会把这个op报给相关部门支持，要等未来版本出来大家才能用到。
2. 自己修改模型绕过这个 ...

好的，感谢， 下一个发布版本会支持嘛？ 下个版本预计什么时候发布哈?
另外， 如果是tensorflow 实现的这个模型，  PixelShuffle  算子 rknn 目前支持不？