解决RuntimeError: Exporting the operator deform_conv2d to ONNX opset version 14 is not supported. Please feel free to request support or submit a pull request on PyTorch GitHub问题

参考

一、安装对应的包

pip install deform_conv2d_onnx_exporter

二、构建简易网络模型

import torch
import torchvision.ops
from torch import nn
import deform_conv2d_onnx_exporter
deform_conv2d_onnx_exporter.register_deform_conv2d_onnx_op()

class mynet(nn.Module):
    def __init__(self):
        super(mynet,self).__init__()
        self.weight = torch.rand(5, 3, 3, 3)  # (out_channels, in_c