mmdeploy/configs/_base_/torch2onnx.py
2021-06-17 15:28:23 +08:00

3 lines
98 B
Python

pytorch2onnx = dict(
export_params=True, keep_initializers_as_inputs=False, opset_version=11)