Remove side effects of disabling gradient computaiton (#13257)

0e4f7270 · Lysandre Debut · GitHub · b1198a84 · 0e4f7270
Unverified Commit 0e4f7270 authored Aug 25, 2021 by Lysandre Debut Committed by GitHub Aug 25, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

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

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