Skip to content

Save output tensors in bert_test_data tool#6872

Merged
tianleiwu merged 2 commits intomasterfrom
tlwu/bert_test_data
Mar 4, 2021
Merged

Save output tensors in bert_test_data tool#6872
tianleiwu merged 2 commits intomasterfrom
tlwu/bert_test_data

Conversation

@tianleiwu
Copy link
Contributor

Description:

Allow bert_test_data.py to generate test data for BERT model for onnx_test_runner to verify correctness.

Motivation and Context

  • Why is this change required? What problem does it solve?
  • If it fixes an open issue, please link to the issue here.

onnx_test_runner need output tensors to verify the correctness of model.

@tianleiwu tianleiwu requested a review from a team as a code owner March 3, 2021 04:09
@tianleiwu tianleiwu requested a review from wangyems March 3, 2021 04:10
@tianleiwu tianleiwu changed the title Allow save output tensors for bert_test_data tool Save output tensors in bert_test_data tool Mar 3, 2021
@tianleiwu tianleiwu requested review from liuziyue and yufenglee March 4, 2021 05:53
@tianleiwu tianleiwu merged commit 8f1786d into master Mar 4, 2021
@tianleiwu tianleiwu deleted the tlwu/bert_test_data branch March 4, 2021 21:09
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants