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
1edafeca
allow benchmark_gpt2 on user's model
efee40e0
refactor: get_dummy_inputs returns a data class.
0407a852
tianleiwu
changed the title WIP: Export GPT-2 ONNX model without postion_ids and attention_mask inputs Export GPT-2 ONNX model without postion_ids and attention_mask inputs 5 years ago
fix a type declartion
9e79346c
gh-yewang
approved these changes
on 2020-08-20
tianleiwu
merged
268d2283
into master 5 years ago
tianleiwu
deleted the tlwu/gpt2_model_no_padding branch 5 years ago
Login to write a write a comment.
Login via GitHub