diff --git a/torch/onnx/__init__.py b/torch/onnx/__init__.py index eb081ddf733..00e288bf4b0 100644 --- a/torch/onnx/__init__.py +++ b/torch/onnx/__init__.py @@ -186,7 +186,7 @@ def export(model, args, f, export_params=True, verbose=False, training=TrainingM opset_version (int, default 9): The version of the `default (ai.onnx) opset `_ to target. Must be >= 7 and <= 15. - do_constant_folding (bool, default False): Apply the constant-folding optimization. + do_constant_folding (bool, default True): Apply the constant-folding optimization. Constant-folding will replace some of the ops that have all constant inputs with pre-computed constant nodes. dynamic_axes (dict> or dict, default empty dict):