Flax summarization script (#12230)

* add summrization script * fix arguments, preprocessing, metrics * add generation and metrics * auto model, prediction loop * prettify * label smoothing * adress Sylvain and Patricks suggestions * dynamically import shift_tokens_right * fix shift_tokens_right_fn call

Flax summarization script (#12230)
* add summrization script * fix arguments, preprocessing, metrics * add generation and metrics * auto model, prediction loop * prettify * label smoothing * adress Sylvain and Patricks suggestions * dynamically import shift_tokens_right * fix shift_tokens_right_fn call
c0fe3c9a · Suraj Patil · GitHub · 26a2e365 · c0fe3c9a
Unverified Commit c0fe3c9a authored Jun 23, 2021 by Suraj Patil Committed by GitHub Jun 23, 2021
Expand all Hide whitespace changes
Inline Side-by-side

Showing with 797 additions and 0 deletions

examples/flax/summarization/run_summarization_flax.py examples/flax/summarization/run_summarization_flax.py +797 -0

No files found.
--- a/examples/flax/summarization/run_summarization_flax.py
+++ b/examples/flax/summarization/run_summarization_flax.py