Grad enabled typo

46554fc1 · Lysandre · 0e4f7270 · 46554fc1
Commit 46554fc1 authored Aug 25, 2021 by Lysandre
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/onnx/convert.py src/transformers/onnx/convert.py +1 -1

No files found.
--- a/src/transformers/onnx/convert.py
+++ b/src/transformers/onnx/convert.py
@@ -127,7 +127,7 @@ def export(
    )
    config.restore_ops()
-    torch.set_grad_enabled(False)
+    torch.set_grad_enabled(True)
    return matched_inputs, onnx_outputs