riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-02 03:01:07 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Grad enabled typo

Browse Source
This commit is contained in:
Lysandre 2021-08-25 11:39:45 +02:00
parent 0e4f727069
commit 46554fc12f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/onnx/convert.py
View File

@ -127,7 +127,7 @@ def export(
) )
config.restore_ops() config.restore_ops()
torch.set_grad_enabled(False) torch.set_grad_enabled(True)
return matched_inputs, onnx_outputs return matched_inputs, onnx_outputs