onnxruntime
268d2283
- Export GPT-2 ONNX model without postion_ids and attention_mask inputs (#4852)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
5 years ago
Export GPT-2 ONNX model without postion_ids and attention_mask inputs (#4852) * Export GPT-2 ONNX model without postion_ids and attention_mask inputs * allow benchmark_gpt2 on user's model * refactor: get_dummy_inputs returns a data class.
References
#4852 - Export GPT-2 ONNX model without postion_ids and attention_mask inputs
Author
tianleiwu
Parents
26546f81
Loading