【pytorch】——报错 exporting the operator pixel_unshuffle to onnx opset version 11 is not supported
·
pytorch, onnx
问题：
pytorch的模型转onnx的时候，报错：
exporting the operator pixel_unshuffle to onnx opset version 11 is not supported
但是用pytorch.jit.trace则是可以导出的。
解决方案
- 自己写一个pixel_unshuffle导出到onnx的插件
- 使用pytorch jit.trace追踪出来的模型。
更多推荐
0
0- 0
已为社区贡献22条内容
所有评论(0)