onnxruntime
25885cf7 - Add option --torchscript in benchmark_gpt2.py (#4500)

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
Author
Parents
Loading