pytorch2onnx = dict(
    export_params=True,
    keep_initializers_as_inputs=False,
    opset_version=11,
    save_file='torch2onnx.onnx')