10 lines
230 B
Python
10 lines
230 B
Python
|
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)
|