deformconv2d转onnx出现IndexError: Argument passed to at() was not in the map.

环境配置

• cuda11.6
• torch1.13.1+cuda113
• mmcv-full 1.7.1
• mmcv 1.7.1
  在将mmcv实现的deform_conv2d转为onnx的运行过程中出现错误IndexError: Argument passed to at() was not in the map.
  

错误原因

可能是由于torch1.13.1与一些torch2onnx转换算子不匹配导致，可以把torch版本降低些，我这里降到了1.12.1，运行成功。

在此记录。