onnxruntime
Export GPT-2 ONNX model without postion_ids and attention_mask inputs
#4852
Merged

Commits
  • Export GPT-2 ONNX model without postion_ids and attention_mask inputs
    tianleiwu committed 5 years ago
  • allow benchmark_gpt2 on user's model
    tianleiwu committed 5 years ago
  • refactor: get_dummy_inputs returns a data class.
    tianleiwu committed 5 years ago
  • fix a type declartion
    tianleiwu committed 5 years ago
Loading