onnxruntime
25885cf7
- Add option --torchscript in benchmark_gpt2.py (#4500)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
5 years ago
Add option --torchscript in benchmark_gpt2.py (#4500) * support TorchScript * change onnx filename format * change output name prediction_scores to logits
References
#4500 - Add option --torchscript in benchmark_gpt2.py
Author
tianleiwu
Parents
a95ae164
Loading