mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
6 lines
139 B
Python
6 lines
139 B
Python
pytorch2onnx = dict(
|
|
export_params=True,
|
|
keep_initializers_as_inputs=False,
|
|
opset_version=11,
|
|
save_file='torch2onnx.onnx')
|