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,运行成功。

在此记录。

猜你喜欢

转载自blog.csdn.net/weixin_45453121/article/details/130288012