examples/flax/language-modeling/run_t5_mlm_flax.py · 4212bb0d60b63f81cc2ef44efec02b2203c65eed · chenpangpang / transformers

"test/vscode:/vscode.git/clone" did not exist on "6f1f4b59e35d1c251ba6945d3b562735eaa93203"

[Re-submit] Compute true loss Flax examples (#19504) · 4212bb0d

Duong A. Nguyen authored Oct 13, 2022



* Compute true loss

* fixup

* final

* final

* final

* Update examples/flax/language-modeling/run_bart_dlm_flax.py
Co-authored-by: Sanchit Gandhi <93869735+sanchit-gandhi@users.noreply.github.com>

* jax.tree_map => jax.tree_util.tree_map

* Compute true loss

* final

* fixup

* final

* final

* Update examples/flax/language-modeling/run_bart_dlm_flax.py
Co-authored-by: Sanchit Gandhi <93869735+sanchit-gandhi@users.noreply.github.com>

* jax.tree_map => jax.tree_util.tree_map
Co-authored-by: Sanchit Gandhi <93869735+sanchit-gandhi@users.noreply.github.com>

4212bb0d

run_t5_mlm_flax.py 43.1 KB

Replace run_t5_mlm_flax.py