在将PyTorch模型导出成ONNX时,需要注册额外的符号函数
import mmcv from mmcv.onnx import register_extra_symbolics opset_version = 11 register_extra_symbolics(opset_version)