Grad enabled typo

This commit is contained in:
Lysandre 2021-08-25 11:39:45 +02:00
parent 0e4f727069
commit 46554fc12f

View File

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