mmdeploy/configs/_base_/torch2onnx.py

mirror of https://github.com/open-mmlab/mmdeploy.git synced 2025-01-14 08:09:43 +08:00
pytorch2onnx = dict(
    export_params=True,
    keep_initializers_as_inputs=False,
    opset_version=11,
    save_file='torch2onnx.onnx')
add configs, add convert pipeline 2021-06-17 15:28:23 +08:00			`pytorch2onnx = dict(`
add tensorrt support (#2) 2021-06-23 13:14:28 +08:00			`export_params=True,`
			`keep_initializers_as_inputs=False,`
			`opset_version=11,`
			`save_file='torch2onnx.onnx')`