onnx_config.py 249 Bytes
Newer Older
limm's avatar
limm committed
1
2
3
4
5
6
7
8
9
10
onnx_config = dict(
    type='onnx',
    export_params=True,
    keep_initializers_as_inputs=False,
    opset_version=11,
    save_file='end2end.onnx',
    input_names=['input'],
    output_names=['output'],
    input_shape=None,
    optimize=True)