Skip to content

Latest commit

 

History

History
19 lines (12 loc) · 315 Bytes

onnx.md

File metadata and controls

19 lines (12 loc) · 315 Bytes

MMCV中ONNX模块简介 (实验性)

register_extra_symbolics

在将PyTorch模型导出成ONNX时,需要注册额外的符号函数

范例

import mmcv
from mmcv.onnx import register_extra_symbolics

opset_version = 11
register_extra_symbolics(opset_version)

常见问题