pytorch2onnx = dict(
    export_params=True, keep_initializers_as_inputs=False, opset_version=11)