导出onnx模型,onnx不支持roll操作符

当我torch.onnx.export()函数导出swin-tranformer模型时出现报错:

Exporting the operator roll to ONNX opset version 9 is not supported. Please open a bug to request ONNX export support for the missing operator.

解决方法:

将opset version 9换成12(可以先试试只换opset版本,如果其报错不支持这个版本后再考虑升级pytorch),然后升级pytorch将其有原来的1.7.1版本换成1.11.0后解决问题。

你可能感兴趣的:(深度学习,python,opencv)