Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Update the example of exporting Bart + BeamSearch to ONNX module to resolve comments. #14310

Merged
merged 17 commits into from
Dec 6, 2021
Merged

Update the example of exporting Bart + BeamSearch to ONNX module to resolve comments. #14310

merged 17 commits into from
Dec 6, 2021

Conversation

fatcat-z
Copy link
Contributor

@fatcat-z fatcat-z commented Nov 7, 2021

This is a PR to resolve all of comments left in previous PR:

#13765

Also provides a README.md for this example.

@fatcat-z
Copy link
Contributor Author

fatcat-z commented Nov 7, 2021

@NielsRogge @garymm

Please help to take a review. Thanks.

fatcat-z and others added 5 commits November 9, 2021 00:31
Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
examples/onnx/pytorch/translation/README.md Outdated Show resolved Hide resolved
examples/onnx/pytorch/translation/README.md Outdated Show resolved Hide resolved
examples/onnx/pytorch/translation/README.md Outdated Show resolved Hide resolved
fatcat-z and others added 2 commits November 9, 2021 09:11
@fatcat-z fatcat-z changed the title Update the example of exporting BartModel + BeamSearch to ONNX module to resolve comments. Update the example of exporting Bart + BeamSearch to ONNX module to resolve comments. Nov 9, 2021
@fatcat-z
Copy link
Contributor Author

@michaelbenayoun
Could you please help to take a review? Thanks!

@fatcat-z
Copy link
Contributor Author

@garymm Please help to take a review.

Copy link
Contributor

@garymm garymm left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, very minor comments.
Thanks!

examples/onnx/pytorch/translation/README.md Outdated Show resolved Hide resolved
examples/onnx/pytorch/translation/run_onnx_exporter.py Outdated Show resolved Hide resolved
fatcat-z and others added 2 commits November 16, 2021 11:16
Co-authored-by: Gary Miguel <garymm@garymm.org>
Co-authored-by: Gary Miguel <garymm@garymm.org>
Copy link
Member

@lewtun lewtun left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This is really elegant work @fatcat-z - thank you for making the code much cleaner!

I left a few minor comments, mostly about trying to make things a bit clearer to the end-user. Otherwise it looks good to me!

@lewtun
Copy link
Member

lewtun commented Dec 6, 2021

Thanks for integrating all the feedback @fatcat-z ! Is it OK if I merge this?

@lewtun lewtun merged commit 1ccc033 into huggingface:master Dec 6, 2021
@fatcat-z fatcat-z deleted the bart_bs_to_onnx_update branch December 6, 2021 16:03
Albertobegue pushed a commit to Albertobegue/transformers that referenced this pull request Jan 27, 2022
…esolve comments. (huggingface#14310)

* Update code to resolve comments left in previous PR.

* Add README.md file for this example.

* Update examples/onnx/pytorch/translation/README.md

Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>

* Update examples/onnx/pytorch/translation/README.md

Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>

* Update examples/onnx/pytorch/translation/README.md

Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>

* Update README.md file to resolve comments.

* Add a section name.

* Update examples/onnx/pytorch/translation/README.md

Co-authored-by: Gary Miguel <garymm@garymm.org>

* Add more comments for _convert_past_list_to_tuple().

* Change the default file name to a consistent one.

* Fix a format issue.

* Update examples/onnx/pytorch/translation/README.md

Co-authored-by: Gary Miguel <garymm@garymm.org>

* Update examples/onnx/pytorch/translation/run_onnx_exporter.py

Co-authored-by: Gary Miguel <garymm@garymm.org>

* Update examples/onnx/pytorch/translation/README.md

Co-authored-by: lewtun <lewis.c.tunstall@gmail.com>

* Change the folder to summarization and address some other coments.

* Update the torch version.

Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
Co-authored-by: Gary Miguel <garymm@garymm.org>
Co-authored-by: lewtun <lewis.c.tunstall@gmail.com>
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.

None yet

4 participants